洛谷 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;
}