P2383 狗哥玩木棒 约翰玩木棒

P2383 狗哥玩木棒 约翰玩木棒

题目背景
狗哥又趁着语文课干些无聊的事了…

题目描述
现给出一些木棒长度，那么狗哥能否用给出的木棒（木棒全用完）组成一个正方形呢？

输入格式：

输入文件中的第一行是一个整数n表示测试的组数，接下来n行表示每组的测试数据。 每行的第一个数为m(4<=m<=20),接下来m个数ai(1<=ai<=1000)表示木棒的长度。

输出格式：

对于每组测试数据，如果可以组成正方形输出“yes”，否则输出“no”。

输入输出样例
输入样例：
3
4 1 1 1 1
5 10 20 30 40 50
8 1 7 2 6 4 4 3 5
输出样例：
yes
no
yes

//【参考代码】
#include 
#include 
using namespace std;
const int N = 1000;
int n, m;
int s[N];
int eval;//边长  平均数
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>eval||b>eval||c>eval||d>eval)
    {
        return;//其中一条边大于平均数 长木棒不能变短
    }
    if(a==eval&&b==eval&&c==eval&&d==eval)
    {
        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);
    while(n--)
    {
        int all = 0;//求所有的木头之和
        memset(s,0,sizeof(s));
        scanf("%d",&m);
        for(int i=1; i<=m; i++)
        {
            scanf("%d",&s[i]);
            all+=s[i];
        }
        //正方形的4条边是相等的  不整除4不可能是正方形
        if(all%4!=0)
        {
            puts("No");
            continue;
        }
        eval = all/4;//正方形的每一条边
        sort(s+1,s+m+1,cmp);//从大到小进行排序
        dfs(1,0,0,0,0);
        if(flag==1)  puts("yes");
        else puts("no");
    }

    return 0;
}