正方形 dfs

计蒜客 正方形 （dfs好题，具有参考价值）

阅读数：473

题目：

题目思路：先将木棍长度数组降序排序，dfs前先判断一下是否符合题意，然后进行dfs，dfs出结果就退出，需要剪枝的地方，dfs 用一个参数记录上一次搜索的位置在哪里，下一次搜索只会在这条边之后。搜索完一条边后pos再置为0。

代码：

1. #include<bits/stdc++.h>
2. using namespace std;
3. int l[ 30];
4. bool mark[ 30];
5. int n,sum,one;
6. bool cmp(int a,int b)
7. {
8. return a>b;
9. }
10. int flag= 0;
11. void dfs(int onel,int num,int pos)
12. {
13. if(flag== 1) return ;
14. if(num== 3){
15. flag= 1;
16. return ;
17. }
18. if(onel == 0){
19. num++;
20. onel=one;
21. pos= 0;
22. }
24. for( int i=pos;i<n;i++){
25. if(!mark[i]){
26. if(onel>=l[i]){
27. mark[i]= true; //标记为已访问
28. dfs(onel-l[i],num,i);
29. mark[i]= false; //未访问
30. }
31. }
32. }
33. }
35. int main()
36. {
37. fill(mark,mark+ 30, false);
38. cin>>n;
39. int sum= 0;
40. for( int i= 0;i<n;i++){
41. cin>>l[i];
42. sum+=l[i];
43. }
44. one = sum/ 4;
45. sort(l,l+n,cmp);
46. if(l[ 0]>one||sum% 4) cout<< "No"<< endl;
47. else{
48. dfs(one, 0, 0);
49. if(flag== 1) cout<< "Yes"<< endl;
50. else cout<< "No"<< endl;
51. }
52. }