HDU-1518-square

#include<iostream>using namespace std;#define max(a,b) (a)>(b)?(a):(b)int a[23],sum,aver,n,maxn;bool b[23],flag;void dfs(int mysum,int count,int j){    if(mysum==aver)    {        count++;        if(count==4)            {                flag=true;                return ;            }        else            mysum=0;            j=0;    }    if(flag) return ;//一旦能围城正方形立即跳出，防止TLE    for(int i=j;i<n;i++)        if(!b[i]&&mysum+a[i]<=aver)        {            b[i]=true;            dfs(mysum+a[i],count,i);            b[i]=false;        }}int main(){    int k,i;    cin>>k;    while(k--)    {      cin>>n;        sum=0;        maxn=0;        flag=false;        for(i=0;i<n;i++)            {                cin>>a[i];                maxn=max(maxn,a[i]);                sum+=a[i];            }        aver=sum/4;        if(sum%4!=0||maxn>aver)//简单判断能不能围成一个正方形          {  flag=false;}        else            dfs(0,0,0);        if(flag)            cout<<"yes"<<endl;        else            cout<<"no"<<endl;    }        return 0;}