#include <bits/stdc++.h> using namespace std; int main() { int n; cin >> n; long long z = 0; vector<long long> v(n * n + 1, 0); for (int a = 1; a <= n; a++) { for (int b = a; b <= n; b++) { int x = a * a + b * b; if (x <= n * n) { v[x]++; } } } for (int d = 1; d <= n; d++) { int d2 = d * d; for (int h = 1; h < d; h++) { int y = d2 - h * h; if (y > 0) z += v[y]; } } cout << z; 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 | #include <bits/stdc++.h> using namespace std; int main() { int n; cin >> n; long long z = 0; vector<long long> v(n * n + 1, 0); for (int a = 1; a <= n; a++) { for (int b = a; b <= n; b++) { int x = a * a + b * b; if (x <= n * n) { v[x]++; } } } for (int d = 1; d <= n; d++) { int d2 = d * d; for (int h = 1; h < d; h++) { int y = d2 - h * h; if (y > 0) z += v[y]; } } cout << z; return 0; } |