#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'; } |