#include<bits/stdc++.h>
#define LL long long
#define LLL __int128
#define uint unsigned
#define ldb long double
#define uLL unsigned long long
using namespace std;
const int N=5e7+5;
int n,ans;
int f[N],g[N];
signed main(){
cin.tie(0)->sync_with_stdio(0);
cin>>n;
for(int i=1;i<=n;++i)for(int j=1;j<=i;++j)++f[i*i+j*j],++g[i*i-j*j];
for(int i=1;i<=n*n;++i)ans+=1ll*f[i]*g[i];
cout<<ans<<'\n';
return 0;
}
/*
*/
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | #include<bits/stdc++.h> #define LL long long #define LLL __int128 #define uint unsigned #define ldb long double #define uLL unsigned long long using namespace std; const int N=5e7+5; int n,ans; int f[N],g[N]; signed main(){ cin.tie(0)->sync_with_stdio(0); cin>>n; for(int i=1;i<=n;++i)for(int j=1;j<=i;++j)++f[i*i+j*j],++g[i*i-j*j]; for(int i=1;i<=n*n;++i)ans+=1ll*f[i]*g[i]; cout<<ans<<'\n'; return 0; } /* */ |
English