#include "bits/stdc++.h"
using namespace std;
#define all(x) x.begin(),x.end()
template<typename A, typename B> ostream& operator<<(ostream &os, const pair<A, B> &p) { return os << p.first << " " << p.second; }
template<typename T_container, typename T = typename enable_if<!is_same<T_container, string>::value, typename T_container::value_type>::type> ostream& operator<<(ostream &os, const T_container &v) { string sep; for (const T &x : v) os << sep << x, sep = " "; return os; }
#ifdef LOCAL
#include "debug.h"
#else
#define debug(...) 42
#define ASSERT(...) 42
#endif
typedef long long ll;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef pair<int,int> pi;
const int oo = 1e9;
const int N = 5000;
int ans[N*N+1];
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int n; cin >> n;
// a^2 + b^2 + h^2 == n^2
for(int a=1;a<=n;++a) for(int b=a;b<=n;++b) if(a*a + b*b <= n*n) {
ans[a*a+b*b]++;
}
ll res=0;
for(int h=1;h<=n;++h) {
for(int j=h;j<=n;++j)
res+=ans[j*j-h*h];
}
cout << res << '\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 | #include "bits/stdc++.h" using namespace std; #define all(x) x.begin(),x.end() template<typename A, typename B> ostream& operator<<(ostream &os, const pair<A, B> &p) { return os << p.first << " " << p.second; } template<typename T_container, typename T = typename enable_if<!is_same<T_container, string>::value, typename T_container::value_type>::type> ostream& operator<<(ostream &os, const T_container &v) { string sep; for (const T &x : v) os << sep << x, sep = " "; return os; } #ifdef LOCAL #include "debug.h" #else #define debug(...) 42 #define ASSERT(...) 42 #endif typedef long long ll; typedef vector<int> vi; typedef vector<vi> vvi; typedef pair<int,int> pi; const int oo = 1e9; const int N = 5000; int ans[N*N+1]; int main() { ios_base::sync_with_stdio(false); cin.tie(NULL); int n; cin >> n; // a^2 + b^2 + h^2 == n^2 for(int a=1;a<=n;++a) for(int b=a;b<=n;++b) if(a*a + b*b <= n*n) { ans[a*a+b*b]++; } ll res=0; for(int h=1;h<=n;++h) { for(int j=h;j<=n;++j) res+=ans[j*j-h*h]; } cout << res << '\n'; } |
English