code Tổng của các số chính phương bằng C++

#include <iostream>
#include <fstream>
#include <cmath>
using namespace std;
bool ok(int x) {
    int s=sqrt(x);
    return s*s==x;
}
int main() {
    ifstream input("SUMOF.INP");
    ofstream output("SUMOF.OUT");
    int n;
    input>>n;
    int d=0;
    for (int a=1;a*a<=n;++a) {
        for (int b=a;a*a+b*b<=n;++b) {
            int c=n-a*a-b*b;
            if (c>=b*b && ok(c)) {
                ++d;
            }
        }
    }
    output<<d<<endl;
    input.close();
    output.close();
    return 0;
} 

#danglam228

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll n,d=0;
int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    freopen("SUMOF.INP","r",stdin);
    freopen("SUMOF.OUT","w",stdout);
    cin >> n;
    for (ll i = 1; i * i <= n; ++i) {
        ll d1 = i * i;
        for (ll j = i; d1 + j * j <= n; ++j) {
            ll d2 = j * j;
            ll d3 = n - d1 - d2;
            ll k = sqrt(d3);
            if (k >= j && k * k == d3) {
                ++d;
            }
        }
    }
    cout << d;
    return 0;
}