深搜（dfs）（拼正方形）

给定一些长度各异的木棒，请问是否可能把它们连接成一个正方形？

输入

第一行含有一个数t，代表数据组数。
每组数据一行，开头的一个数m，代表有m个木棍。接下来m个数分别是每根木棒的长度。
4<=m<=20，木棒长度在1到10000之间。

输出

每组数据一行，如果能拼出一个正方形输出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<bits/stdc++.h>
using namespace std;
int n,sum;
int a[55];
bool vis[55];
 
bool dfs(int cur, int st,int res) {
    if(cur == 4) return 1;
    if(res < 0) return 0 ;
    if(res == 0) {
        int bg = 1;
        while(vis[bg]) ++bg;
        for(int i = bg; i<=n; i++) {
            if(vis[i]) continue;
            vis[i]=1; 
            if(dfs(cur+1,i+1,sum/4-a[i])) return 1;
            vis[i]=0;
        }
        return 0;
    }
//    if(st > n) return 0;
    for(int i = st; i<=n; i++) {
        if(vis[i]) continue;
        vis[i]=1;
        if(dfs(cur,i+1,res - a[i])) return 1;
        vis[i]=0;
    }
    return 0;
}
bool cmp(const int & a,const int & b) {
    return a > b;
}
int main()
{
    int t;
    cin>>t;
    while(t--) {
        scanf("%d",&n);
        sum=0;
        int flag=1;
        for(int i = 1; i<=n; i++) scanf("%d",a+i),sum+=a[i]; 
        if(sum%4 != 0 ){
            printf("no\n");continue;
        }
        memset(vis,0,sizeof vis);
        sort(a+1,a+n+1,cmp);
        if(dfs(1,1,sum/4)) printf("yes\n");
        else printf("no\n");
    }
    return 0 ;
}