洛谷——P2383 狗哥玩木棒

P2383 狗哥玩木棒

题目背景

狗哥又趁着语文课干些无聊的事了...

题目描述

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

输入输出格式

输入格式：

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

输出格式：

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

输入输出样例

输入样例#1：

3
4 1 1 1 1
5 10 20 30 40 50
8 1 7 2 6 4 4 3 5

输出样例#1：

yes
no
yes

说明

狗哥快抓狂了

搜索+特判

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 100
using namespace std;
bool flag;
];
int cmp(int a,int b){return a>b;}
int read()
{
    ,f=;char ch=getchar();
    ;ch=getchar();}
    +ch-',ch=getchar();
    return x*f;
}
void dfs(int x)
{
    if(flag) return ;
    )
    {
        flag=;
        return ;
    }
    ;i<=;i++)
    {
        if(a[x]>p[i]) continue;
        p[i]-=a[x];
        dfs(x+);
        p[i]+=a[x];
    }
}
int main()
{
    Q=read();
    while(Q--)
    {
        n=read(),tot=;
        ;i<=n;i++) a[i]=read(),tot+=a[i];
        !=) {printf("no\n"); continue;}
        sort(a+,a++n,cmp);
        ]>tot/) {printf("no\n"); continue;}
        ;i<=;i++) p[i]=tot/;
        flag=);
        if(flag) printf("yes\n");
        else printf("no\n");
    }
    ;
}