POJ 2362 square &&POJ 1011 Sticks 记忆化搜索 DFS

最新推荐文章于 2021-12-08 08:58:35 发布

kdqzzxxcc

最新推荐文章于 2021-12-08 08:58:35 发布

阅读量730

点赞数

分类专栏： DFS

本文链接：https://blog.csdn.net/just_water/article/details/7843337

版权

DFS 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

poj2362

题意：给出一些木棒，问你是否能将这些木棒组成一个正方形。

思路:

1.当木棒总长度不被4整除则输出no。

2.可以这样理解，将所有木棒分成4组等长的，和为sum/4的木棒，搜索的到则成立。

3.如果失败则回溯到上一个状态，若一直回溯到第一个状态，则找不到。

我用了一个visit[]数组来保存木棒是否已经被取用,

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cmath>
#include <cstring>
#include <queue>
#include <set>
#include <vector>
#include <stack>
#include <map>
#include <iomanip>
#define PI acos(-1.0)
#define Max 2005
#define inf 1<<28
using namespace std;

int sticks[Max],sum,n;
bool visit[Max];
int length;

int dfs(int duan, int start, int left)//left代表剩余的length长度
{
    int i;
    if(left == 0)// 如果left为0，证明正好组成一条变长为length的木棒
    {
        //cout<<duan<<endl<<endl;
        //show();
        if(duan == 3) return 1;//如果前面已经搜索到3段了，则证明这是第四段，则返回成功。
        for(i=0; i<n; i++)//找出未被取到的木棒的最小的序数，从最小的序数继续DFS
            if(!visit[i])
                break;
        //cout<<i<<":"<<endl;
        visit[i] = 1;//标记为找过
        if(dfs(duan + 1, i + 1, length - sticks[i]))//如果成功，则返回成功
            return 1;
        visit[i] = 0;//否则不取这一段
        return 0;//回溯
    }
    //cout<<start<<endl;
    if(start >= n) return 0;
    for(int j = start; j < n; j++)//从start开始搜索
    {
        if(visit[j] || sticks[j] > left)
            continue;
        //cout<<j<<endl;
        visit[j] = 1;
        if(dfs(duan, j +1, left - sticks[j])) return 1;
        visit[j] = 0;
    }
    return 0;
}

int main()
{
    int i,j,k,l,m,T;
    cin>>T;
    while(T--)
    {
        cin>>n;
        sum=0;
        memset(visit,0,sizeof(visit));
        for(i=0; i<n; i++)
        {
            scanf("%d",&sticks[i]);
            sum+=sticks[i];
        }
        if(sum%4!=0)
        {
            cout<<"no"<<endl;
            continue;
        }
        length=sum/4;
        int flag=dfs(0,0,length);//从第0段，序数0，length剩余长度开始搜索
        if(flag)
            cout<<"yes"<<endl;
        else
            cout<<"no"<<endl;
    }
    return 0;
}

poj1011

其实原理和上面那题一样，就是2362那题木棒的数量已经给定为4，而这题木棒的数量没有限制，只要找出组合相同的最小长度即可。

当然这一题很重要的就是一些特殊情况的舍弃，我就没考虑一些情况TLE了好几次。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cmath>
#include <cstring>
#include <queue>
#include <set>
#include <vector>
#include <stack>
#include <map>
#include <iomanip>
#define PI acos(-1.0)
#define Max 2005
#define inf 1<<28
using namespace std;

int sticks[Max];
bool visit[Max];
int n,m,num;

bool cmp(int x,int y)//注意排序 ，从大到小排，自己理解一下，递归的次数，从小到大排就TLE
{
    return x>y;
}
int dfs(int no,int num1,int zong)//搜索和上面那题差不多，就是return 1 的条件不同而已
{
    int i,j;
    if(num1==0)
    {
        if(zong==0)return 1;
        for(i=0; i<n; i++)
            if(!visit[i])
                break;
        visit[i]=1;
        if(dfs(i+1,num-sticks[i],zong-sticks[i]))
            return 1;
        visit[i]=0;
        return 0;
    }
    if(no>=n)return 0;
    for(i=no; i<n; i++)
    {
        if(visit[i]||sticks[i]>num1)
            continue;
        if(i > 0 && !visit[i - 1] && sticks[i] == sticks[i - 1])//如果该木棒和前面木棒的长度相同，则不选
            continue;
        visit[i]=1;
        if(dfs(i+1,num1-sticks[i],zong-sticks[i]))
            return 1;
        visit[i]=0;
    }
    return 0;
}
int main()
{
    int i,j,k,l;
    while(scanf("%d",&n),n)
    {
        int sum=0,MAX=-inf;
        for(i=0; i<n; i++)
        {
            visit[i]=0;
            scanf("%d",&sticks[i]);
            sum+=sticks[i];
            if(sticks[i]>MAX)
                MAX=sticks[i];
        }
        sort(sticks,sticks+n,cmp);
        for(i=MAX; i<=sum; i++)//枚举原始木棒的长度，从单一木棒的最大值枚举到所有木棒的总长度
        {
            if(sum%i!=0)//剪枝，如果枚举的木棒长度不被总长度整除，则肯定不是原始木棒的长度
            continue;
            num=i;
            if(dfs(0,i,sum))
            {
                printf("%d\n",i);
                break;
            }
        }

    }
    return 0;
}

这两题做下来，相对来说1011要更多的考虑剪枝的情况，不然很容易TLE。