#include <bits/stdc++.h> #define debug(x) cout<<#x<<" = "<<x<<"\n" #define pb push_back #define ins insert #define fi first #define se second using namespace std; using ull = unsigned long long; using ll = long long; using ld = long double; template <typename H, typename T> ostream& operator<<(ostream& os, pair<H, T> m){ return os <<"("<< m.fi<<", "<<m.se<<")"; } template <typename H> ostream& operator<<(ostream& os, vector<H> V){ os<<"{"; for(int i=0; i<V.size(); i++){ if(i)os<<" "; os<<V[i]; } os<<"}"; return os; } void solve(); int main(){ ios_base::sync_with_stdio(false); cin.tie(NULL); int t = 1; //cin>>t; while(t--) solve(); return 0; } vector<int> pierw[25000000+1]; void solve(){ for(int i = 1; i <= 5000; i++){ for(int j = 1; j < i; j++){ pierw[i*i-j*j].pb(j); } } int n; int cnt = 0; cin>>n; int n2 = n*n; for(int h = 1; h <= n; h++){ for(int a = 1; a <= n; a++){ int u = h*h+a*a; if(u > n2) break; for(auto x : pierw[u]){ int z = u+x*x; if(x <= a && z <= n2) cnt++; } } if(h*h > n2) break; } cout<<cnt<<"\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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 | #include <bits/stdc++.h> #define debug(x) cout<<#x<<" = "<<x<<"\n" #define pb push_back #define ins insert #define fi first #define se second using namespace std; using ull = unsigned long long; using ll = long long; using ld = long double; template <typename H, typename T> ostream& operator<<(ostream& os, pair<H, T> m){ return os <<"("<< m.fi<<", "<<m.se<<")"; } template <typename H> ostream& operator<<(ostream& os, vector<H> V){ os<<"{"; for(int i=0; i<V.size(); i++){ if(i)os<<" "; os<<V[i]; } os<<"}"; return os; } void solve(); int main(){ ios_base::sync_with_stdio(false); cin.tie(NULL); int t = 1; //cin>>t; while(t--) solve(); return 0; } vector<int> pierw[25000000+1]; void solve(){ for(int i = 1; i <= 5000; i++){ for(int j = 1; j < i; j++){ pierw[i*i-j*j].pb(j); } } int n; int cnt = 0; cin>>n; int n2 = n*n; for(int h = 1; h <= n; h++){ for(int a = 1; a <= n; a++){ int u = h*h+a*a; if(u > n2) break; for(auto x : pierw[u]){ int z = u+x*x; if(x <= a && z <= n2) cnt++; } } if(h*h > n2) break; } cout<<cnt<<"\n"; } |