poj 2362 hdoj 1518 Square(搜索)

题目链接

大致题意:

给定一堆不定长度的小棒子,问他们能否构成一个正方形。

 

解题思路:

POJ1011的热身题,DFS+剪枝

 

本题大致做法就是对所有小棒子长度求和sum,sum就是正方形的周长,sum/4就是边长side。

问题就转变为:这堆小棒子能否刚好组合成为4根长度均为side的大棒子

 

不难了解,小棒子的长度越长,其灵活性越差。例如长度为5的一根棒子的组合方式要比5根长度为1的棒子的组合方式少,这就是灵活性的体现。

由此,我们首先要对这堆小棒子降序排序,从最长的棒子开始进行DFS

 

剪枝,有3处可剪:

1、  要组合为正方形,必须满足sum%4==0;

2、  所有小棒子中最长的一根,必须满足Max_length <= side,这是因为小棒子不能折断;

3、  当满足条件1、2时,只需要能组合3条边,就能确定这堆棒子可以组合为正方形。


#include 
#include 
#include 
using namespace std;

bool vis[21];
int stick[21];
int n;
int side;

bool cmp(int a, int b)
{
    return a > b;
}

bool dfs(int num, int len, int s)//s为搜索起点
{
    if (num == 4)
        return true;
    for (int i = s; i < n; i++)
    {
        if (vis[i])
            continue;
        vis[i] = true;
        if (len+stick[i] < side)
        {
            if (dfs(num, len+stick[i], i))
            //这里解释游戏为什么有搜索起点这一变量,只有这用到了,如果len+stick[i] > side了
            //因为是非升序,len+stick[i-1] > side必然成立,从0开始就是浪费时间
                return true;
            /*开始这里我直接 return dfs(num, len+stick[i], i);貌似这样也可以,
            但的的确确是错的,直接返回会导致我们少判很多种情况*/
        }
        else if (len+stick[i] == side)
        {
            if (dfs(num+1, 0, 0))
                return true;
            //这里同上
        }
        vis[i] = false;
    }
    return  false;
}
int main()
{
    int t;
    scanf("%d", &t);
    while (t--)
    {
        scanf("%d", &n);
        int sum = 0;
        for (int i = 0; i < n; i++)
        {
            scanf("%d", &stick[i]);
            sum += stick[i];
        }
        side = sum/4;
        sort(stick, stick+n, cmp);
        if (sum % 4 || side < stick[0])
        //这里也可以去掉一些情况
        {
            puts("no");
            continue;
        }
        memset(vis, false, sizeof(vis));
        if (dfs(1, 0, 0))
            puts("yes");
        else
            puts("no");
    }
    return 0;
}


转载于:https://www.cnblogs.com/xindoo/archive/2013/04/29/3595100.html

你可能感兴趣的:(poj 2362 hdoj 1518 Square(搜索))