洛谷 P2383 狗哥玩木棒【dfs】
洛谷 P2383 狗哥玩木棒
https://www.luogu.org/problemnew/show/P2383
题目大意:
给出n个木棒的长度,问这些木棒能否拼凑成为一个正方形
题目分析:
这个题目,首先是要拼成正方形,所以,我们可以确定每一条边的长度应该是总长度除以4。然后,我们就进行搜索,由于我们知道每一条边的长度,当某一条边上的值大于答案,那么这个答案就不需要考虑了。就可以对这种情况进行剪枝。
代码
#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+50;
int n;
int m,eva;
int s[maxn];
int flag;
bool cmp(int a,int b){return a>b;}
void dfs(int k,int a,int b,int c,int d)
{
if(a>eva||b>eva||c>eva||d>eva)return;
if(a==eva&&b==eva&&c==eva&&d==eva) flag=1;
if(flag==1)return;
dfs(k+1,a+s[k],b,c,d);
dfs(k+1,a,b+s[k],c,d);
dfs(k+1,a,b,c+s[k],d);
dfs(k+1,a,b,c,d+s[k]);
}
int main()
{
scanf("%d",&n);
int all=0;
while(n--)
{
scanf("%d",&m);
all=0;
memset(s,0,sizeof(s));
for(int i=1;i<=m;i++) {scanf("%d",&s[i]);all+=s[i];}
if(all%4) {puts("no");continue;}
eva=all/4;
sort(s+1,s+1+m,cmp);
flag=0;
dfs(1,0,0,0,0);
if(flag)puts("yes");
else puts("no");
}
return 0;
}