用python验证四方定理

题目

菲菲视频讲解

举例验证

请输入一个自然数:483

(0, 1, 121, 361)

请输入一个自然数:5739

(0, 25, 1225, 4489)

请输入一个自然数:4803

(0, 1, 2401, 2401)

代码

importitertools

n =int(input("请输入一个自然数:"))

l = []

foriinrange(,int(n **0.5) +1):

l.append(i **2)

forjinitertools.product(l, repeat =4):

ifsum(j) == n:

print(j)

break