问题遇到的现象和发生背景
编写函数,把一个自然数分解成最多4个平方数的和,要求越短越好。
问题相关代码,请勿粘贴截图
from itertools import combinations
from random import sample
def short(data):
pingfangshu=[1]
def generate():
n=max(data)
for num in range(2,int(n**0.5)+1):
for _ in range(4):
pingfangshu.append(num**2)
generate()
def getresult(num):
temp=int(num**0.5)**2
index=pingfangshu.index(temp)+1
temppingfangshu=pingfangshu[:index]
for length in range(1,5):
for item in combinations(pingfangshu,length):
if sum(item)==num:
return item
for num in data:
print(num,':',getresult(num))
data=sample(range(1,100),10)
short(data)
运行结果及报错内容
generate这个函数是怎么定义的?这个函数的作用是什么?