pku 2362 Square

这是我在csdn写的第一篇文章，这道题也是我第一次用回溯法做出来的题目，虽然不是什么难题，但是挺有纪念意义，以此记载

题目意思：

给出你不同长短的木棍，问是否能够构成一个正方形

解题思路：

暴力搜索+简单剪枝

1、如果所有木棍的长度加起来不能整除4，那么一定不能构成正方形

2、如果有一条木棍长度超过一条边的长度，那么也可以排除了

3、搜索前按降序排序能减少重复搜索

#include <iostream> #include <cstdlib> using namespace std; int cmp(const void *a, const void *b) { return *(int *)b - *(int *)a; } int list[21],n,sum,average; bool visited[21],flag; void backtrack(int edge, int s, int len) { if(edge <= 1) flag = true; if(flag == true) return; if(len==0) // 如果这木棍能够形成一条边 for(int i = 1; i <= n; ++i) if(!visited[i] && flag == false) { visited[i] = true; backtrack(edge-1, i, average - list[i]); visited[i] = false; } for(int i = s; i <= n; ++i) // 还不能形成一条边 if(!visited[i] && len >= list[i] && flag == false) { visited[i] = true; backtrack(edge, i, len - list[i]); visited[i] = false; } } void dfs() { flag = false; for(int i = 1; i <= n; ++i) visited[i] = false; backtrack(4, 1, average); } int main() { int t,sum,rs; cin>>t; while(t--) { cin>>n; flag = true; sum = 0; for(int i = 1; i <= n; ++i) { cin>>list[i]; sum += list[i]; } if(sum%4==0) average = sum/4; else // 如果所有数加起来不能整除4就不用管了 flag = false; qsort(&list[1], n, sizeof(int), cmp); if(list[1] > average) flag = false; // 如果有一条木棍大于平均值也不构正方形 if(flag) { dfs(); } if(flag) cout<<"yes"<<endl; else cout<<"no"<<endl; } return 0; }