思路:
构造check函数,检查我们当前的排序是否满足题目中所给的条件,在深度搜索时调用check函数进行剪枝。
在深度搜索时,生成前num个数的全排列(见注释),每生成一次都进行check检查,避免浪费时间。
代码:
ans=0 b=[0 for _ in range(15)] vis=[0 for _ in range(15)] def check3(): return b[1]+b[2]==b[3] def check6(): return b[4]-b[5]==b[6] def check9(): return b[7]*b[8]==b[9] def check12(): return b[10]==b[11]*b[12] def dfs(num): global ans if num==13: if check12(): ans+=1 return if num==4 and not check3():return if num==7 and not check6():return if num==10 and not check9():return for i in range(1,14): if not vis[i]: #如果i没有被访问过,则访问i vis[i]=1 #标记访问过i了,b[]中之后不再出现i b[num]=i #赋值 dfs(num+1) #寻找当前b[]的下一个值 vis[i]=0 #回退,清理标记,寻找下一组 dfs(1) print(ans)