《python算法训练4-枚举》代码参考

□ □ □+ □ □ □= □ □ □，将1~9分别填入9个□，每个数字只能使用一次使得等式成立。

方法一

for a in range(1,10):
    for b in range(1,10):
        for c in range(1,10):
            for d in range(1,10):
                for e in range(1,10):
                    for f in range(1,10):
                        for g in range(1,10):
                            for h in range(1,10):
                                for i in range(1,10):
                                    flag=[0 for k in range(10)]
                                    flag[a]=1
                                    flag[b]=1
                                    flag[c]=1
                                    flag[d]=1
                                    flag[e]=1
                                    flag[f]=1
                                    flag[g]=1
                                    flag[h]=1
                                    flag[i]=1
                                    sum=0
                                    for k in range(10):
                                        sum=sum+flag[k]
                                        
                                    if(100*a+10*b+c+100*d+10*e+f==100*g+10*h+i)and(sum==9):
                                        print(a,b,c,d,e,f,g,h,i)

方法二

import itertools
s=[i for i in range(1,10)]
num=list(itertools.permutations(s,9))

for i in num:
    if(i[0]*100+i[1]*10+i[2] + i[3]*100+i[4]*10+i[5]==i[6]*100+i[7]*10+i[8]):
        print(i)