#pragma GCC optimize("O3")
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define ld double
#define pb push_back
#define ff first
#define ss second
#define MOD 1000000009
#define INF 1000000019
#define INFL 1000000000000000099LL
int zlicz[50000000][2];
ll ans,n;
int main()
{
ios_base::sync_with_stdio(0);cin.tie(0);
cin>>n;
for(ll i=1;i<=n;i++){
for(ll j=i;i*i+j*j<=n*n;j++){
zlicz[i*i+j*j][0]++;
}
}
for(ll i=1;i<=n;i++){
for(ll j=1;j<i;j++){
zlicz[i*i-j*j][1]++;
}
}
for(ll i=0;i<n*n+7;i++){
ans+=zlicz[i][0]*zlicz[i][1];
}
cout<<ans;
return 0;
}
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 | #pragma GCC optimize("O3") #include <bits/stdc++.h> using namespace std; #define ll long long #define ld double #define pb push_back #define ff first #define ss second #define MOD 1000000009 #define INF 1000000019 #define INFL 1000000000000000099LL int zlicz[50000000][2]; ll ans,n; int main() { ios_base::sync_with_stdio(0);cin.tie(0); cin>>n; for(ll i=1;i<=n;i++){ for(ll j=i;i*i+j*j<=n*n;j++){ zlicz[i*i+j*j][0]++; } } for(ll i=1;i<=n;i++){ for(ll j=1;j<i;j++){ zlicz[i*i-j*j][1]++; } } for(ll i=0;i<n*n+7;i++){ ans+=zlicz[i][0]*zlicz[i][1]; } cout<<ans; return 0; } |
English