[蓝桥杯] 数字划分 （Python 实现）

题目：

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

---

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

代码：

import itertools
number=[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16]
n=list(itertools.combinations(number,8))
for a in n:
    a=set(a)
    x=set(number)^a
    if sum(a)==sum(x):
        r1,r2=1,1
        for b1 in a:
            r1+=(b1**2)
        for b2 in x:
            r2+=(b2**2)
        if r1==r2:
            r3,r4=1,1
            for c1 in a:
                r3+=(c1**3)
            for c2 in x:
                r4+=(c2**3)
            if r3==r4:
                print(a,x)

结果：1, 4, 6, 7, 10, 11, 13, 16