给定一些长度各异的木棒,请问是否可能把它们连接成一个正方形?
输入
第一行含有一个数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 ;
}