1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
#include <iostream>
#include <vector>
using namespace std;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int n;
    cin>>n;
    long long res = 0;

    vector<int> plus(n*n,0), minus(n*n,0);
    for (int a=1; a<n; a++) for (int b=a; a*a+b*b<n*n; b++) plus[a*a+b*b]++;
    for (int h=1; h<n; h++) for (int d=h+1; d<=n; d++) minus[d*d-h*h]++;
    for (int i=1; i<n*n; i++) res += plus[i]*minus[i];

    cout<<res<<'\n';

    return 0;
}