很明显的一道深搜,给你一些线段,看是否能拼成正方形
先找到总长度,总长度不能整除4就直接输出NO
找到边长
在搜的过程中,出来当前值大于边长 等于边长小于边长的情况,以及何时回溯的问题
注意 DFS的3个参数的一个 K来标记数组下标,当当前值等于边长的时候 要从开头开始遍历,这跟回溯不同
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<queue>
#include<vector>
#include<map>
#include<string>
#define inf 0x7fffffff
#define maxn 200
using namespace std;
int n;
int cnt;
int res;
int flag=0;
int hash[30];
int a[30];
void dfs(int id,int sum,int k)
{
if(id==n)
{
flag=1;
return;
}
if(flag)
return;
for(int i=k;i<n;i++)
{
if(hash[i])
continue;
int t=sum+a[i];
if(t>res)
continue;
hash[i]=1;
if(t==res)
dfs(id+1,0,0);
if(t<res)
dfs(id+1,t,i+1);
hash[i]=0;
}
}
int main() {
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
memset(hash,0,sizeof(hash));
int i;
int sum=0;
for(i=0;i<n;i++)
{
scanf("%d",&a[i]);
sum+=a[i];
}
if(sum%4!=0)
printf("no\n");
else
{
flag=0;
res=sum/4;
dfs(0,0,0);
if(flag)
printf("yes\n");
else
printf("no\n");
}
}
return 0;
}