Advertisement
dsuveges

Untitled

Jan 19th, 2023 (edited)
955
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.46 KB | None | 0 0
  1. import math
  2. def find_all_squares(n):
  3.     squares = set()
  4.     for i in range(1, int(math.sqrt(n)) + 1):
  5.         a = i**2
  6.         for j in range(i, int(math.sqrt(n)) + 1):
  7.             b = j**2
  8.             c = n - a - b
  9.             if c >= 0 and c == int(c) and math.isqrt(c)**2 == c:
  10.                 squares.add(tuple(sorted([a, b, c])))
  11.     if squares:
  12.         return [list(sq) for sq in squares]
  13.     else:
  14.         return None
  15.  
  16. print(find_all_squares(51))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement