POJ_2362_Square(DFS)

题型：搜索

题意：将n个木棍组成一个正方形

分析：

与1011的Sticks类似，DFS+剪枝，关键在于如下剪枝：

(1)若总长度不能被4整除，则无法组成正方形

(2)若最长的木棍比预期的正方形的边长还要长，则无法组成正方形

(3)在组合的过程中就发现无法构成一条符合的边长，则没有必要再继续搜索了

有了这些剪枝，写起来就比较清晰简单了。

代码：

#include<iostream>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<algorithm>
using namespace std;
int n,a[50];
bool vis[50],flag;
int standard;
int sum,max_stick;
bool cmp(int a,int b) {
return a>b;
}

void dfs(int begin,int len,int cnt) {
if(cnt==3) {
flag = true;
return;
}
if(len==standard) dfs(0,0,cnt+1);
if(flag) return;
for(int i=begin; i<n; i++) {
if(!vis[i] && a[i]+len<=standard) {
vis[i] = true;
dfs(i+1,len+a[i],cnt);
if(flag) return;
vis[i] = false;
}
}

}

int main() {
int t;
scanf("%d",&t);
while(t--) {
sum=0;
max_stick=0;
scanf("%d",&n);
for(int i=0; i<n; i++) {
scanf("%d",&a[i]);
sum+=a[i];
max_stick=max(max_stick,a[i]);
}
if(sum%4!=0 || max_stick*4>sum) {
printf("no\n");
continue;
}
standard=sum/4;
sort(a,a+n,cmp);
memset(vis,false,sizeof(vis));
flag=false;
dfs(0,0,0);
if(flag) printf("yes\n");
else printf("no\n");
}
return 0;
}