day44 数字划分

数字划分

w星球的长老交给小明一个任务：1,2,3...16 这16个数字分为两组。
要求：这两组数字的和相同，并且，两组数字的平方和也相同，并且，两组数字的立方和也相同。
请你利用计算机的强大搜索能力解决这个问题。
并提交1所在的那个分组的所有数字。
这些数字要从小到大排列，两个数字间用一个空格分开。
即类似：1 4 5 8 ...  这样的答案。
注意，只提交这一组数字，不要填写任何多余的内容。

----------------------------------------
笨笨有话说：
    只要一个组的成员确定了，另一个组的成员也就确定了。枚举一个组的成员就可以了。
	凭直觉，两个组的成员数目不会差太多吧。
歪歪有话说：
    既然求 1 所在的那个组，那只要枚举剩余的成员就可以了。
	貌似都是8个成员的可能性很大啊。

from itertools import combinations

lst = [i for i in range(1, 17)]
sum1 = sum(lst)
sum2 = sum([i ** 2 for i in lst])
sum3 = sum([i ** 3 for i in lst])

res = [1]
for possible in combinations(lst[1:], 7):
    if sum(possible) + 1 == sum1 / 2:
        if sum([i ** 2 for i in possible]) + 1 == sum2 / 2:
            if sum([i ** 3 for i in possible]) + 1 == sum3 / 2:
                res.extend(list(possible))
                print(' '.join(map(str, res)))