// clang-format off
#include <bits/stdc++.h>
using namespace std;
#define ALL(x) (x).begin(), (x).end()
template<class C> void mini(C &a5, C b5) { a5 = min(a5, b5); }
template<class C> void maxi(C &a5, C b5) { a5 = max(a5, b5); }
#ifdef LOCAL
const bool debug = true;
#else
const bool debug = false;
#define cerr if (true) {} else cout
#endif
// clang-format on
#define int long long
#define double long double
const int INF = 1e18;
const int mod = 1e9 + 7;
const int nax = 5000;
int ab_count[nax * nax + 10];
void init() {
for (int a = 1; a <= nax; a++) {
for (int b = a; b <= nax; b++) {
int ab = a * a + b * b;
if (ab > nax * nax) {
break;
}
ab_count[ab]++;
}
}
}
int32_t main() {
ios_base::sync_with_stdio(0);
cin.tie(0);
init();
int N;
cin >> N;
int res = 0;
for (int h = 1; h <= nax; h++) {
for (int n = 1; n <= N; n++) {
int ab = n * n - h * h;
if (ab < 0) {
continue;
}
res += ab_count[ab];
}
}
cout << res;
return 0;
} /*
*/
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | // clang-format off #include <bits/stdc++.h> using namespace std; #define ALL(x) (x).begin(), (x).end() template<class C> void mini(C &a5, C b5) { a5 = min(a5, b5); } template<class C> void maxi(C &a5, C b5) { a5 = max(a5, b5); } #ifdef LOCAL const bool debug = true; #else const bool debug = false; #define cerr if (true) {} else cout #endif // clang-format on #define int long long #define double long double const int INF = 1e18; const int mod = 1e9 + 7; const int nax = 5000; int ab_count[nax * nax + 10]; void init() { for (int a = 1; a <= nax; a++) { for (int b = a; b <= nax; b++) { int ab = a * a + b * b; if (ab > nax * nax) { break; } ab_count[ab]++; } } } int32_t main() { ios_base::sync_with_stdio(0); cin.tie(0); init(); int N; cin >> N; int res = 0; for (int h = 1; h <= nax; h++) { for (int n = 1; n <= N; n++) { int ab = n * n - h * h; if (ab < 0) { continue; } res += ab_count[ab]; } } cout << res; return 0; } /* */ |
English