问题描述
w星球的长老交给小明一个任务:1,2,3…16 这16个数字分为两组。
要求:这两组数字的和相同,并且,两组数字的平方和也相同,并且,两组数字的立方和也相同。
请你利用计算机的强大搜索能力解决这个问题。
并提交1所在的那个分组的所有数字。
这些数字要从小到大排列,两个数字间用一个空格分开。
即类似:1 4 5 8 … 这样的答案。
注意,只提交这一组数字,不要填写任何多余的内容。
笨笨有话说:
只要一个组的成员确定了,另一个组的成员也就确定了。枚举一个组的成员就可以了。
凭直觉,两个组的成员数目不会差太多吧。
歪歪有话说:
既然求 1 所在的那个组,那只要枚举剩余的成员就可以了。
貌似都是8个成员的可能性很大啊。
思路分析及代码实现
import itertools
li = [i for i in range(1, 17)]
total = sum(li)
total2 = 0
total3 = 0
for j in li:
total2 += j*j
for k in li:
total3 += k*k*k
n = list(itertools.combinations(li, 8))
for i in n:
sum2 = 0
sum3 = 0
if sum(i) == total//2:
for m in i:
sum2 += m*m
for n in i:
sum3 += n*n*n
if sum2 == total2//2 and sum3 == total3//2:
if 1 in i:
for q in i:
print(q, end=' ')