HJ89 24点运算 用递归生成器进行数列穷举

思路:

1、对4张牌进行全排序，并输出列表

2、分别对排序进行计算尝试，采用穷举方式

3、返回结果

除了用递归生成器进行数组全排序外，也用模块from itertools import permutations，进行全排序。

#输出算式运算顺序从左至右运算，不需要括号确定优先级。

#列举所有排序方式，比如 1234，1324......
def relist(a):#用递归生成器返回不重复顺序的列表
    if len(a)==1:
        yield a
    for i in range(len(a)):
        for j in relist(a[:i]+a[i+1:]):
            #print( a[i],j)
            yield [a[i]]+j   

def twentyfour(a):#采用多重循环对数组a所有计算方式穷举。
    operator=['+','-','*','/']
    for i in operator:
        for j in operator:
            for n in operator:
                if dop[i](dop[j](dop[n](a[0],a[1]),a[2]),a[3])==24:
                    return a,n,j,i
import operator
dop={'+':operator.add,'-':operator.sub,'*':operator.mul,'/':operator.truediv}
d={'J':11,'Q':12,'K':13,'A':1}
a=input().split()
b=a.copy()
flag=0
for k,i in enumerate(a):
    if i.isdigit():
        a[k]=int(i)
    elif i in d:
        a[k]=d[i]
    else:
        print("ERROR")
        flag=1
#用flag使当输出”ERROR"不再进行后面运算，用flag1,输出NONE.
if flag==0:
    flag1=0
    for i in relist(a):
        #print(i)
        new=[]
        if twentyfour(i):
            temp,n1,n2,n3=twentyfour(i)
            for i in temp:
                for k,j in enumerate(a):
                    if i==j:                             #查表打印特殊字符JQKA
                        new.append(b[k])
                        break
            tem=str(new[0])+n1+str(new[1])+n2+str(new[2])+n3+str(new[3])
            print(tem)
            flag1=1
            break
    if flag1==0:
        print('NONE')