import math
def count_valid_aquariums(n):
count = 0
max_sq = n * n
is_square = [False] * (max_sq + 1)
for i in range(1, int(math.sqrt(max_sq)) + 1):
is_square[i * i] = True
for d in range(1, n + 1):
d_sq = d * d
for a in range(1, d):
a_sq = a * a
for b in range(a, d):
ab_sq = a_sq + b * b
if ab_sq >= d_sq:
break
h_sq = d_sq - ab_sq
if h_sq > 0 and is_square[h_sq]:
count += 1
return count
print(count_valid_aquariums(int(input())))
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 | import math def count_valid_aquariums(n): count = 0 max_sq = n * n is_square = [False] * (max_sq + 1) for i in range(1, int(math.sqrt(max_sq)) + 1): is_square[i * i] = True for d in range(1, n + 1): d_sq = d * d for a in range(1, d): a_sq = a * a for b in range(a, d): ab_sq = a_sq + b * b if ab_sq >= d_sq: break h_sq = d_sq - ab_sq if h_sq > 0 and is_square[h_sq]: count += 1 return count print(count_valid_aquariums(int(input()))) |
English