FJUT 1320 正方形

第二天，小教练来到了学校，老师考了他一些问题发现他都能答对，就又想出一个题目考他。
老师给他一些木棍，然后问它能否拼出一个正方形。 当然这还是需要你们的帮助，请你们写好程序，让叮当狗发到教练手机上。

(必须用到所有木棍)

Input
第一个输入案例个数 N。每个样例包含木棍数目M， 4 <= M <= 20。然后输入M个木棍的长度。长度是个整数并且不超过10000。
Output
如果可以拼成正方形输出 “yes” 否则输出 “no”。
SampleInput
3
4 1 1 1 1
5 10 20 30 40 50
8 1 7 2 6 4 4 3 5
SampleOutput
yes
no
yes

首先能围成正方形，并且要用到所有木棍，得保证木棍长度之和能被4整除。如果不能，则输出no，能则进行搜索。
能被4整除，最后每条边的长度一定就是总和除以4（因为所有木棍都要用到）。然后就是在这m根木棍中，任意选取组合让和为avg，并且有4个组合。

#include<bits/stdc++.h>
using namespace std;
int a[25],m,flag,v[25];
/// t表示到了第几个木棍，sum表示组合之和，ss是满足组合之和等于avg个数，avg是平均值
int dfs(int t,int sum,int ss,int avg)
{
     if(sum==avg)///选取的木棍之和等于平均值
     {
         ss++;///组合个数加一
         t=1,sum=0;///从第一开始搜，并且组合之和重新开始所以为0；
         if(ss==4)///当满足4个组合时候 说明可以组成正方形
         {
             flag=1;
            return 1;
         }
     }
     for(int i=t; i<=m; i++)
     {
         if(!v[i]&&sum+a[i]<=avg)///v【i】表示是否访问过该木棍
         {
             v[i]=1;
             if(dfs(i+1,sum+a[i],ss,avg))
                 return 1;
             v[i]=0;
         }
     }
     return 0;
}
/*
void dfs(int t,int sum,int ss,int avg)
{
     if(sum==avg)///选取的木棍之和等于平均值
     {
         ss++;///组合个数加一
         t=1,sum=0;///从第一开始搜，并且组合之和重新开始所以为0；
         if(ss==4)///当满足4个组合时候 说明可以组成正方形
         {
             flag=1;
             return ;
         }
     }
     for(int i=t; i<=m; i++)
     {
         if(!v[i]&&sum+a[i]<=avg)///v【i】表示是否访问过该木棍
         {
             v[i]=1;
             dfs(i+1,sum+a[i],ss,avg);
             v[i]=0;
         }
     }
}*/
int main()
{
     int n,i;
     cin>>n;
     while(n--)
     {
         cin>>m;
         int s=0;
         flag=0;
         memset(v,0,sizeof(v));
         for(i=1; i<=m; i++)
             cin>>a[i],s+=a[i];
         if(s%4)
         {
             puts("no");    
         }
         else
         {
             sort(a+1,a+m+1);
             int avg=s/4;
             dfs(1,0,0,avg);
             if(flag)
                 puts("yes");
             else
                 puts("no");
         }
     }
}

最开始用的是void dfs 5000ms TLE了，然后改成int dfs 600多ms ac了。想了下原因，因为如果已经在一个深度中满足了题意，即组成了正方形，int 可以立刻一层一层返回，而用void的话即使组成了正方形，也要遍历dfs所有深度才结束，当木棍个数很多时，递归层数是非常大的。