题意:
给你M根长度可能不同的棍子,问你用着M根棍子能否拼成一个正方形.
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
int sum;
int m;
int a[30];
int vis[30];
int dfs(int cnt,int left,int be) //cnt当前是第几条边,left是这条边还需要多长,be是从第几条边开始搜。
{
if(cnt==4)return 1; //只要有三条边找到了,就完成了。
else for(int i=be;i<=m;i++)if(vis[i]==0)
{
if(a[i]==left)
{
vis[i]=1;
if(dfs(cnt+1,sum/4,1))return 1;
vis[i]=0;
}
if(a[i]<left)
{
vis[i]=1;
if(dfs(cnt,left-a[i],i+1))return 1;//这里为什么从第i+1开始。请看下面分析
vis[i]=0;
}
}
return 0;
}
int main()
{
int t;
cin>>t;
while(t--)
{
int max_len=0;
sum=0;
memset(vis,0,sizeof(vis));
cin>>m;
for(int i=1;i<=m;i++)
{
cin>>a[i];
sum+=a[i];
max_len=max(max_len,a[i]);
}
if(sum%4!=0||max_len>sum/4)
{
cout<<"no"<<endl;
}
else
{
if(dfs(1,sum/4,1))
{
cout<<"yes"<<endl;
}
else
{
cout<<"no"<<endl;
}
}
}
return 0;
}
假设我们当前刚设置完vis[5]=1,表示第5根棍子我们在还是left长度的时候用了,现在我们还是left-len[5]长度的时候要不要去用len[2]来尝试配对一下?不用,肯定无解.因为搜索的时候我们是先找的len[2]的,如果len[2]+len[5]+….有一个可行解(可以配对成当前cnt边)的话,我们之前肯定已经找到了且推出了dfs这个函数.但是现在明显我们还没找到,所以len[2]+len[5]+…不可能是一个可行解,所以直接从5后面的位置继续找即可.如果还不能理解,就把本题想象成我们只需要配对一条正方形的边,不用配4边了.