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