蓝桥杯寒假作业——python

题目如下

现在小学的数学题目也不是那么好玩的。
看看这个寒假作业：

□ + □ = □
□ - □ = □
□ × □ = □
□ ÷ □ = □

(如果显示不出来，可以参见【图1.jpg】)

每个方块代表1~13中的某一个数字，但不能重复。
比如：
6 + 7 = 13
9 - 8 = 1
3 * 4 = 12
10 / 2 = 5

以及：
7 + 6 = 13
9 - 8 = 1
3 * 4 = 12
10 / 2 = 5

就算两种解法。（加法，乘法交换律后算不同的方案）

你一共找到了多少种方案？

请填写表示方案数目的整数。
注意：你提交的应该是一个整数，不要填写任何多余的内容或说明性文字。

题目解析

我尝试了许多方法，首先利用itertools包解决发现循环很长时间没有结束，之后我又尝试利用深度优先搜索，很长时间没出结果

a = [0] * 13
m = [0] * 14
res = 0
def dfs(l):
    global res
    if l == 13:
        q = a[1] + a[2]
        w = a[4] - a[5]
        e = a[7] * a[8]
        r = a[10] / a[11]
        if q == a[3] and w == a[6] and e == a[9] and r == a[12]:
            res += 1
        return
    for i in range(1, 14):
        if m[i] == 0:
            a[l] = i
            m[i] = 1
            dfs(l+1)
            m[i] = 0
dfs(1)
print(res)

没有循环出来，然后看了一下网上的发现了这种方法很快就算出来了

nums = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]
ls = [0]  # 用于计数

#深度优先遍历
def dfs(nums, temp):
    if len(temp) == 3:
        if not check(temp, 3):
            return
    elif len(temp) == 6:
        if not check(temp, 6):
            return
    elif len(temp) == 9:
        if not check(temp,9):
            return
    elif len(temp) == 12:
        if not check(temp,12):
            return
        else:
            ls[0] += 1
            return
    for i in nums:
        new_nums = nums.copy()
        new_nums.remove(i)
        new_temp = temp + [i]
        dfs(new_nums,new_temp)

# 检查函数 检查遍历的
def check(temp, n):
    if n == 3:
        return temp[0] + temp[1] == temp[2]
    elif n == 6:
        return temp[3] - temp[4] == temp[5]
    elif n == 9:
        return temp[6] * temp[7] == temp[8]
    elif n == 12:
        return temp[11] * temp[10] == temp[9]


dfs(nums, [])
print(ls[0])

答案：64