编写函数,把一个自然数分解成最多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这个函数是怎么定义的?这个函数的作用是什么?
1,函数内是可以定义函数的,所以generate在short内定义没错
2,generate的作用:对data内10个数用最大的数生成一个最大限度的能加起来和可能为这个最大数的平方数列表。还有一个4次的循环,目的是有可能平方数有可能重复,但由于题目要求只能是4个数相加,所以这里有了一个range(4)。
3,剩下的就是对这个列表里的元素求组合了。