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
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int main()
{
    unordered_multiset<ll>s;
    ll n;
    cin>>n;
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<i; j++)
            s.insert(i*i-j*j);
    }
    ll res=0;
    for(int i=1; i<=n; i++)
    {
        for(int j=i; j<=n; j++)
        {
            //cout<<i<<" "<<j<<endl;
            if(s.find(i*i+j*j)!=s.end())
            {
                res+=s.count(i*i+j*j);
                //cout<<i<<" "<<j<<endl;
            }
        }
    }
    cout<<res;
}