hdu 1518 square解题报告

/*此题刚看时真没思路，搜了下别人的解题报告才有思路，还是用了个多小时

题意：给出很多棍子，问能否正好拼出个正方形

此题用回溯，一个数一个数的搜，如果之和为正好为给出所以数总数的1/4，则边数加一；

*/#include<iostream>

using namespace std;

int n,m;

int sum,len;

int a[22],v[22];

int ok;

void  dfs(int cur,int l,int b)//cur 第几条边，l 当前边的长度，b,数组中第几个，其实可以不要这个，但会超时，因为访问过的就不必再访问

{
if(l==sum/4)
dfs(cur+1,0,0);
if(cur==4)
{
ok=1;
return ;
//cout<<"yes"<<endl;
//break;
}
if(ok==1)//保证能跳出
return ;
if(cur!=4)
{
for(int i=b;i<n;i++)
if(!v[i]&&l+a[i]<=sum/4)
{
v[i]=1;
dfs(cur,l+a[i],i);
v[i]=0;
}
}

}

int main()

{
int t,i,j;
cin>>t;
while(t--)
{
cin>>n;
sum=0;
len=0;
for(i=0;i<n;i++)
{
cin>>a[i];
sum+=a[i];
ok=0;
if(a[i]>len)
len=a[i];
}
if(len>sum/4||sum%4!=0)//小小的剪枝
cout<<"no"<<endl;
else
{
memset(v,0,sizeof(v));
dfs(0,0,0);
if(ok)
cout<<"yes"<<endl;
else
cout<<"no"<<endl;

}
}
return 0;

}