Python递归穷举解决24点问题

问题描述：在一副扑克牌去掉大小王后，随机抽取4张牌，对其进行四则运算（可带括号），寻找结果等于24的解法
解决方案：对牌的顺序、括号位置、可能的运算符穷举，直到找到解法
知识点：递归思想，python中排列组合、求笛卡尔积的函数
局限性：指数级时间复杂度，仅适合少量数据(牌的张数较少)

代码(python 3.7)

import random
import itertools

# 假定仅4张牌
def find24(cards):
    #   可用的运算符
    ops = ['+', '-', '*', '/']
    print('计算牌：' + ','.join([str(i) for i in cards]))

    #   所有计算符号的排列组合
    opsSeq = [i for i in itertools.permutations(ops, 3)]
    #    所有数字的排列组合
    cardsSeq = [i for i in itertools.permutations(cards, 4)]
    for seqCard in cardsSeq:
        for ps in opsSeq:
            #           将尝试的数字组合加上括号
            numbersWithBraces = addBrace([str(i) for i in seqCard])
            for numBraces in numbersWithBraces:
                #               将加上括号的数字，和运算符组合，组成最终表达式
                express = ''.join(list(itertools.chain.from_iterable(zip(numBraces, ps))) + [numBraces[3]])
                try:
                    #                   计算表达式的值
                    if eval(express) == 24:
                        print('找到解法' + express + '=24')
                        return
                except:
                    continue
    print('没有找到解法')


# 递归给数字加上所有可能的括号组合
def addBrace(numbers):
    if len(numbers) < 2: return [numbers]
    if len(numbers) == 2:
        return [['(' + str(numbers[0])] + numbers[1:-1] + [str(numbers[-1]) + ')']]

    results = []
    for i in range(1, len(numbers)):
        prefixs = addBrace(numbers[:i])
        tails = addBrace(numbers[i:])

        for p, t in itertools.product(prefixs, tails):#求笛卡尔积
            brace_with_around = ['(' + p[0]] + p[1:] + t[:-1] + [t[-1] + ')']
            results.append(brace_with_around)

    return results

# 随机四张牌
card1 = random.sample(range(1, 13), 4)
find24(card1)