首页 > 其他 > 详细

Square HDU 1518 搜索

时间:2019-10-11 21:15:03      阅读:76      评论:0      收藏:0      [点我收藏+]

Square HDU 1518 搜索

题意

原题链接

给你一定若干个木棒,让你使用它们组成一个四边形,要求这些木棒必须全部使用。

解题思路

木棒有多种组合方式,使用搜索来进行寻找,这里需要进行优化,不然复杂度非常高。

因为M最大为20,所以,如果用DFS还是可以接受的。

由所有火柴棒的长度和,我们可以求出要组成正方形的边长。我们设边长为len。在搜索前,首先可以把边长为分数的,也就是火柴棒的长度和不能被4整除的,排除掉。

问题可以转化为从M个数中挑出4组和为len的数。

首先,从M个数中搜索出和为len的数。如果能搜到,则接着从剩余的数中,进行同样的搜索,依此类推,一共进行4段这样的搜索。

如果某一段搜索失败,则回溯到上一段搜索。

如果一直回溯到第一段的搜索,仍未能找到解,则说明无解。

搜索过程

将长度按从大到小排序。

对应正方形的4条边,进行4段同样的搜索。而且这4段搜索也是递归式的,就是说,当第一段边搜索到后,便递归进行下一段的搜索。由此可见,搜索过程中要用到的信息包括:当前是第几段;当前段还剩余的长度;搜索的起始位置。

每段搜索的过程是相同的:因为到最后,所有的火柴棒都是会被用到的,所以,每当开始一个新段的搜索时,总是把剩余的第一个数放入,在这段的搜索过程中,它始终是放入状态。然后进入堆栈的下一层,试着放入下一个剩余的数,如果放入这个数后递归到下面未回溯回来,则说明已经找到最后答案,程序结束。如果得到回溯,那么把该数取出,同理再试着放入排在它后面的数。

搜索的起始位置

起始位置就是,在每次进入递归函数的时候,应该从哪个位置的数开始试着放入;如果这次递归要放入的是一个段的第一个数,这个数肯定是剩余的第一个数,而且这个数是不会被取出的,也就是说,这一层的可选状态只有一个,如果放入这个数之后,后面发现无法搜索到一个完整的边,那么说明剩余的数到达最终的目标,必须回溯到上一个段。

如果这次递归要放入不是一个段的第一个数,那么它的起始位置肯定是在它上一层递归放入的数的后面,但是它前面可能还存在没有被放入的数,这些数就不用再试了吗?

因为是对同一个段的搜索,所以,组成这个段的数被放入的顺序是没有关系的。所以,如果一个数作为一个段的第一个数放入不合适,那么它作为第二个数或第三个数放入肯定也是不合适的。

代码实现

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=1e4+7;
int a[maxn], vis[maxn], sum;
int m, len;

bool dfs(int duan, int begin, int left)//记录当前的段数,数组的位置,当前段还需要多长的木棍
{
    if(left==0) 
    {
        if(duan==3) return 1;
        int i;
        for( i=1; i<=m && vis[i]!=0; i++);
        vis[i]=1;
        if(dfs(duan+1, i, len-a[i])) return 1;
        vis[i]=0;
        return 0;
    }
    for(int i=begin; i<=m; i++)
    {
        if(vis[i]==1 || a[i] > left) continue;
        if(i>1 && vis[i-1]==0 && a[i-1]==a[i]) continue;
        vis[i]=1;
        if(dfs(duan, i, left-a[i])) return 1;
        vis[i]=0;
    }
    return 0;
}
bool cmp(int a, int b)
{
    return a>b;
}
int main()
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d", &m);
        sum=0;
        for(int i=1; i<=m; i++)
        {
            scanf("%d", &a[i]);
            sum+=a[i];
            vis[i]=0;
        }
        if(sum%4!=0)
        {
            printf("no\n"); 
            continue;
        } 
        sort(a+1, a+1+m, cmp);
        len=sum/4;
        if(a[1] > len)
        {
            printf("no\n");
            continue;
        }
        vis[1]=1;
        if( dfs(1, 1, len-a[1]) )
            printf("yes\n");
        else printf("no\n"); 
    }
    return 0;
 } 

Square HDU 1518 搜索

原文:https://www.cnblogs.com/alking1001/p/11656687.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!