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
#include <bits/stdc++.h>
using namespace std;

const int MAXN = 5000;
vector<int> sumy(MAXN *MAXN + 1);

int main() {
  ios_base::sync_with_stdio(0);
  cin.tie(0);

  int n;
  cin >> n;

  for (int a = 1; a <= n; a++) {
    for (int b = a; b <= n; b++) {
      int kw = a * a + b * b;
      if (kw <= MAXN * MAXN) {
        sumy[kw]++;
      }
    }
  }

  long long wynik = 0;

  for (int h = 1; h <= n; h++) {
    int h2 = h * h;
    for (int przek = h + 1; przek <= n; przek++) {
      int przek2 = przek * przek;
      int kw = przek2 - h2;
      if (kw > 0 && kw <= MAXN * MAXN) {
        wynik += sumy[kw];
      }
    }
  }

  cout << wynik << '\n';
  return 0;
}