import math def count_valid_aquariums(n): count = 0 for a in range(1, n + 1): for b in range(a, n + 1): # b >= a, aby uniknąć duplikatów for h in range(1, n + 1): diag_sq = a * a + b * b + h * h diag = math.isqrt(diag_sq) if diag * diag == diag_sq and diag <= n: count += 1 return count if __name__ == "__main__": n = int(input().strip()) print(count_valid_aquariums(n))
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | import math def count_valid_aquariums(n): count = 0 for a in range(1, n + 1): for b in range(a, n + 1): # b >= a, aby uniknąć duplikatów for h in range(1, n + 1): diag_sq = a * a + b * b + h * h diag = math.isqrt(diag_sq) if diag * diag == diag_sq and diag <= n: count += 1 return count if __name__ == "__main__": n = int(input().strip()) print(count_valid_aquariums(n)) |