if __name__ == '__main__': n = int(input()) k = 0 sq_table = set() for i in range(n + 1): sq_table.add(i*i) n_sq = n*n for h in range(1, n + 1): for a in range(1, n + 1): for b in range(a, n + 1): diag_squared = a * a + b * b + h * h if diag_squared <= n_sq and diag_squared in sq_table: k += 1 print(k)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | if __name__ == '__main__': n = int(input()) k = 0 sq_table = set() for i in range(n + 1): sq_table.add(i*i) n_sq = n*n for h in range(1, n + 1): for a in range(1, n + 1): for b in range(a, n + 1): diag_squared = a * a + b * b + h * h if diag_squared <= n_sq and diag_squared in sq_table: k += 1 print(k) |