#include <bits/stdc++.h>
using namespace std;
#define _DEBUG
#ifdef _DEBUG
template<typename T1, typename T2> auto& operator<<(ostream& o, pair<T1, T2> a) { return o << "(" << a.first << ", " << a.second << ")"; }
template<typename T, typename OS> auto& operator<<(OS& o, T a) { o << "{"; for (auto b : a) o << b << ", "; return o << "}"; }
#define dbg(x...) cerr << "[" #x "]: ", [](auto... args) { ((cerr << args << ", "),...) << "\n"; }(x)
#else
#define dbg(...)
#endif
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(), (x).end()
#define F first
#define S second
using ll = long long;
using ld = long double;
using pll = pair<ll, ll>;
using vi = vector<int>;
int main() {
ios::sync_with_stdio(false); cin.tie(nullptr);
ll n;
cin >> n;
vector<ll> cnt(n*n+1);
for(ll a = 1 ; a <= n ; ++a) {
for(ll b = a ; b <= n ; ++b) {
if(a*a + b*b <= n*n) {
cnt[a*a + b*b]++;
}
}
}
ll ans = 0;
for(ll d = 1 ; d <= n ; ++d) {
for(ll h = 1 ; h < d ; ++h) {
ans += cnt[d*d - h*h];
}
}
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 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | #include <bits/stdc++.h> using namespace std; #define _DEBUG #ifdef _DEBUG template<typename T1, typename T2> auto& operator<<(ostream& o, pair<T1, T2> a) { return o << "(" << a.first << ", " << a.second << ")"; } template<typename T, typename OS> auto& operator<<(OS& o, T a) { o << "{"; for (auto b : a) o << b << ", "; return o << "}"; } #define dbg(x...) cerr << "[" #x "]: ", [](auto... args) { ((cerr << args << ", "),...) << "\n"; }(x) #else #define dbg(...) #endif #define sz(x) ((int)(x).size()) #define all(x) (x).begin(), (x).end() #define F first #define S second using ll = long long; using ld = long double; using pll = pair<ll, ll>; using vi = vector<int>; int main() { ios::sync_with_stdio(false); cin.tie(nullptr); ll n; cin >> n; vector<ll> cnt(n*n+1); for(ll a = 1 ; a <= n ; ++a) { for(ll b = a ; b <= n ; ++b) { if(a*a + b*b <= n*n) { cnt[a*a + b*b]++; } } } ll ans = 0; for(ll d = 1 ; d <= n ; ++d) { for(ll h = 1 ; h < d ; ++h) { ans += cnt[d*d - h*h]; } } cout << ans; return 0; } |
English