POJ 2362 && HDU 1518 Square(dfs)

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;
}

你可能感兴趣的:(POJ 2362 && HDU 1518 Square(dfs))