#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; } /* */ |