#include <bits/stdc++.h>
#define ull long long
#define ll long long
#define pb push_back
#define pf push_front
#define in insert
#define f first
#define s second
#define pii pair<int, int>
#define pib pair<int, bool>
#define cint const int
#define mn2 262144
#define nw '\n'
#define sp " "
#define mn 5009
#define inf 400009
#define ms 25000009
#define int long long
using namespace std;
ull pot[mn], odp, dp[ms];
ull n;
vector<int> v;
int32_t main() {
ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
for(ull i=1; i<=5000; ++i){
for(ull k=i; k<=5000; ++k){
ull sum=i*i+k*k;
if(sum>ms-9) break;
++dp[sum];
}
}
cin>>n;
for(ull i=1; i<=n; ++i){
for(ull j=1; j<i; ++j){
ull b=i*i-j*j;
if(b<2) continue;
if(b>ms-9) continue;
odp+=dp[b];
}
}
cout<<odp;
}
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 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | #include <bits/stdc++.h> #define ull long long #define ll long long #define pb push_back #define pf push_front #define in insert #define f first #define s second #define pii pair<int, int> #define pib pair<int, bool> #define cint const int #define mn2 262144 #define nw '\n' #define sp " " #define mn 5009 #define inf 400009 #define ms 25000009 #define int long long using namespace std; ull pot[mn], odp, dp[ms]; ull n; vector<int> v; int32_t main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); for(ull i=1; i<=5000; ++i){ for(ull k=i; k<=5000; ++k){ ull sum=i*i+k*k; if(sum>ms-9) break; ++dp[sum]; } } cin>>n; for(ull i=1; i<=n; ++i){ for(ull j=1; j<i; ++j){ ull b=i*i-j*j; if(b<2) continue; if(b>ms-9) continue; odp+=dp[b]; } } cout<<odp; } |
English