题意:让你判断一些不同长度的木棍能不能拼成一个正方形的轮廓,
思路:排除特殊情况后dfs搜索
这个题和一般的dfs不同的地方就是,他组成一个正方形后要直接结束dfs而不是继续搜索寻找最值,所以在很多地方都要加上
return指令,
tip:如果不加pos这个变量而是每次都从1开始找的话,会TLE
AC代码:
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int a[32], vis[32];
int flag, sum, tsum, n;
void dfs(int edge,int len,int pos)
{
int i;
if(edge == 3)
{
flag = 1;
return; //这个地方的return返回到上一个dfs,以此类推最后到main函数
}
if(len == tsum)
{
dfs(edge+1,0,0);//从0开始重新搜索
if(flag)
return ;//此处return也不能少
}
for(i = pos; i <=n; i++)
{
if(!vis[i] && len+a[i] <= tsum)
{
vis[i] = 1;
dfs(edge,a[i]+len,i+1);
if(flag)
return;//此处return也不能少
vis[i] = 0;
}
}
}
int main()
{
int t, i;
while(~scanf("%d",&t))
{
while(t--)
{
memset(vis,0,sizeof(vis));
sum = 0, flag = 0;
int maxx=0;
scanf("%d",&n);
for(i = 1; i <= n; i++)
{
scanf("%d",&a[i]);
sum+=a[i];
maxx=max(maxx,a[i]);
}
if(sum%4 != 0)
{
printf("no\n");
continue;
}
tsum = sum/4;
if(maxx>tsum)
{
printf("no\n");
continue;
}
dfs(0,0,1);
if(flag)
printf("yes\n");
else
printf("no\n");
}
}
return 0;
}