题目链接:Square
解题思路:这个题按照一个一个边来找,找到3个就可以了,排序之后只要最大的变边长大于 sum / 4就无法构成正方形,每次找到一根木棍时,要找他的下一根木棍,避免多次重复寻找,刚开始一直TLE,最后看了个别人的博客才发现可以这样剪枝。(DFS的搜索状态量要作为函数参数) 所以发现DFS的题目时要先分析哪些变量跟着搜索深度在变化,之后根据判定条件和题设要求具体求解。
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define MAX 25
using namespace std;
int v[MAX], stick[MAX], m, flag, sum;
void DFS(int x, int len, int k){
if(flag){
return;
}
if(k == 3){
flag = 1;
return;
}
if(len == sum){
DFS(0, 0, k + 1);
}
for(int i = x; i < m; i++){
if(!v[i]){
if(stick[i] + len > sum){
break;
}
else{
v[i] = 1;
DFS(i + 1, len + stick[i], k);
v[i] = 0;
}
}
}
}
int main(){
int i, j, k, t;
scanf("%d", &t);
while(t--){
memset(v, 0, sizeof(v));
sum = flag = 0;
scanf("%d", &m);
for(i = 0; i < m; i++){
scanf("%d", &stick[i]);
sum += stick[i];
}
if(sum % 4 || stick[m - 1] > sum / 4){
printf("no\n");
continue;
}
sum /= 4;
sort(stick, stick + m);
DFS(0, 0, 0);
if(flag){
printf("yes\n");
}
else{
printf("no\n");
}
}
return 0;
}