这道题,要点比较多
1.在深搜之前我们可以剪两次枝:边长和不是4的倍数,或者存在任意一条边比变长和的四分之一要大,直接判否
2.在深搜递归的时候,一旦我们找全了四条边,就可以直接不论什么情况都直接返回就好,如果这时候不加判断条件的话,可能会一直递归到所有的解都完成,肯定会超时,下面的代码中我会注释出来
3.对于边长度直接就是变长和四分之一的边我们直接标注,下次就不用搜索这条边就好
4.在dfs函数中的最后一个judge是有很大的优化的功能的
这个比较难理解,就这么说吧,针对目前的情况,在之前判断不可能构成一条边的初始边,在我们下一次递归的时候,也是必定不会参与构成边的,因为下一 次递归只要没有凑成变长和的四分之一,那么只对上一次递归,len长度是只会少不会增的,好好理解一下,就明白了
AC代码:
#include"iostream"
#include"cstdio"
#include"cstdlib"
#include"cstring"
using namespace std;
bool book[25];
int stick[25];
int sum;
int s;
int flag=0;
void dfs(int have,int len,int judge)
{
if(have==4)
{
flag=1;
return ; //直接递归回去
}
if(len==sum/4)
{
dfs(have+1,0,1);
if(flag==1) return ; //直接递归回去
}
for(int i=judge;i<=s;i++)
{
if(book[i]==0&&len+stick[i]<=sum/4)
{
book[i]=1;
dfs(have,len+stick[i],i+1);
if(flag==1) return ; //直接递归回去,不进行下次的递归,因为已经找到了可行解
book[i]=0;
}
}
}
int main()
{
int t;
cin>>t;
while(t--)
{
flag=0;
sum=0;
s;
cin>>s;
for(int i=1;i<=s;i++)
{
cin>>stick[i];
sum+=stick[i];
}
if(sum%4!=0)
{
cout<<"no"<<endl;
continue;
}
int wflag=0;
for(int i=1;i<=s;i++)
{
if(stick[i]>sum/4)
{
cout<<"no"<<endl;
wflag=1;
break;
}
}
if(wflag==1)
{
cout<<"no"<<endl;
continue;
}
memset(book,0,sizeof(book));
int count=0;
for(int i=1;i<=s;i++)
{
if(stick[i]==sum/4)
{
count++;
book[i]=1;
}
}
dfs(count,0,1);
if(flag==1) cout<<"yes"<<endl;
else cout<<"no"<<endl;
}
return 0;
}