Description
给n个木棒问能否拼成正方形(不许弯折)
Input
第一行为用例组数n,每组用例占一行,第一个整数为木棍根数m,之后m个整数表示木棍长度
Output
对于每组用例,若木棍可以拼成一个正方形则输出yes,否则输出no
Sample Input
3
4 1 1 1 1
5 10 20 30 40 50
8 1 7 2 6 4 4 3 5
Sample Output
yes
no
yes
Solution
对所有木棍长度求和sum,sum就是正方形的周长,sum/4就是边长side。
问题就转变为:这堆木棍能否刚好组合成为4根长度均为side的木棒
不难了解,木棍的长度越长,其灵活性越差。因此,我们首先要对这堆木棍降序排序,从最长的木棍开始进行dfs
剪枝
1,要组合为正方形,必须满足sum%4==0;
2,所有木棍中最长的一根,必须满足Max_length<=side,这是因为木棍不能折断;
3,木棍数量必须不小于4;
4,当满足条件1、2时,只需要能组合3条边,就能确定这堆木棍可以组合为正方形。
Code
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define maxn 25
int n,m,l[maxn],side,sum;
bool flag[maxn];
int cmp(int a,int b)
{
return a>b;
}
bool dfs(int num,int len,int s)//num为已经拼好的边长数,len为正在拼的木棒长度,s为搜索起点
{
if(num==3)//剪枝四
return true;
for(int i=s;i<m;i++)
{
if(flag[i])//该木棍已经被使用
continue;
flag[i]=true;//使用该木棍
if(len+l[i]<side)//可以使用但不足以拼好一条边
{
if(dfs(num,len+l[i],i))
return true;
}
else if(len+l[i]==side)//正好拼好一条边
{
if(dfs(num+1,0,0))
return true;
}
flag[i]=false;//回溯
}
return false;//不能拼成正方形
}
int main()
{
cin>>n;
while(n--)
{
cin>>m;
sum=0;
memset(flag,false,sizeof(flag));//初始化标记数组
for(int i=0;i<m;i++)
{
cin>>l[i];
sum+=l[i];//累加木棍长度
}
side=sum/4;//正方形边长
sort(l,l+m,cmp);//将木棍按长度降序排
if(l[0]>side||sum%4!=0||m<4)//剪枝一,二,三
{
cout<<"no"<<endl;
continue;
}
if(dfs(0,0,0))//能拼成正方形
cout<<"yes"<<endl;
else//不可能拼成正方形
cout<<"no"<<endl;
}
return 0;
}