1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
#include <bits/stdc++.h>
using namespace std;

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int n; cin >> n;
    vector<int> f(n*n+1);
    for (int i = 1; i <= n; i++) {
        for (int j = i; j <= n; j++) {
            if (i*i+j*j <= n*n) f[i*i+j*j]++;
        }
    }
    long long ans = 0;
    for (int d = 1; d <= n; d++) {
        for (int h = 1; h <= d; h++) {
            ans += f[d*d-h*h];
        }
    }
    cout << ans << '\n';
}