#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define fi first
#define se second
const int N=4e5+64;
const int bc=N/64;
int n,k;
int f[5005*5005];
int main(){
ios::sync_with_stdio(false);cin.tie(0);
cin >> n;
for(int a=1; a<=n ;a++){
for(int b=a; b<=n ;b++){
f[a*a+b*b]++;
}
}
ll ans=0;
for(int a=1; a<=n ;a++){
for(int b=a+1; b<=n ;b++){
ans+=f[b*b-a*a];
}
}
cout << ans << '\n';
}
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 | #include<bits/stdc++.h> using namespace std; typedef long long ll; #define fi first #define se second const int N=4e5+64; const int bc=N/64; int n,k; int f[5005*5005]; int main(){ ios::sync_with_stdio(false);cin.tie(0); cin >> n; for(int a=1; a<=n ;a++){ for(int b=a; b<=n ;b++){ f[a*a+b*b]++; } } ll ans=0; for(int a=1; a<=n ;a++){ for(int b=a+1; b<=n ;b++){ ans+=f[b*b-a*a]; } } cout << ans << '\n'; } |
English