Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
25 changes: 25 additions & 0 deletions perfect_square_sum.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
def is_perfect_square(n: int) -> bool:
"""
Check if a number is a perfect square.

:param n: The number to check.
:return: True if the number is a perfect square, False otherwise.
"""
sqrt = n ** 0.5
return sqrt.is_integer()


def perfect_square_sum(numbers: set[int]) -> int:
"""
Calculate the sum of perfect squares that can be formed from the integers in the set.

:param numbers: A set of positive integers.
:return: The sum of perfect squares that can be formed from the input set.
"""
perfect_squares_sum = 0

for num in numbers:
if is_perfect_square(num):
perfect_squares_sum += num

return perfect_squares_sum