码迷,mamicode.com
首页 > 其他好文 > 详细

sicily 1219(记忆化搜索)

时间:2015-06-03 21:43:48      阅读:169      评论:0      收藏:0      [点我收藏+]

标签:

题目链接:sicily 1214

解题思路:
博弈题,用搜索来做。但是,如果用普通的搜索来做的话,是会超时的——复杂度大约是O( n^n ),所以需要采用记忆化搜索的方法(其实差不多就是动态规划了,但是这里是树形DP)。
状态:
用集合S表示现在树的状态,i 表示现在轮到谁进行砍边,dp[ S ][ i ]表示最优值。集合S可以用二进制来表示,即001表示现在还剩下第0条边。
状态转移:
1)A的目标是取最大值,B的目标是取最小值,我们在推导当前状态的最优解时,需要分两种情况考虑!即A得维护较大值,B维护较小值。
2)需要考虑没有边可以砍的情况——在当前玩家无法更新的情况下,直接寻找另一个玩家的解。
预处理:
一切为了时间——题目存在同时砍多条边的情况(只得到离根最近的一条边的值),所以我们可以在搜索前找到每一条边的影响边,即砍掉这条边之后什么边会消失!

代码:

#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
#define INF 0x3f3f3f3f

using namespace std;

int n,c[25][25],w[25][25],v[25];
int dp[1<<20][2],flag[25],pre[25];

void dfs(int index)                     //预处理 
{
    v[index]=1;
    flag[index]|=(1<<(index-1));        //加上自身这个结点(使用后继结点的索引作为边的编号) 
    for(int i=1;i<n;i++)
    {
        if(!v[i]&&w[index][i]!=0)
        {
            dfs(i);
            flag[index]|=flag[i];
            pre[i]=index;
        }
    }
}

int sol(int index,int player)           //记忆化搜索 
{
    if(!index) return 0;
    if(dp[index][(player+1)/2]!=INF)    //已经搜索过的 
        return dp[index][(player+1)/2];

    int maxx=-INF,minx=INF,new_index,tmp;
    for(int i=1;i<n;i++)                //尝试砍去剩下的边 
    {
        if(index&(1<<(i-1)))
        {
            if(player!=c[pre[i]][i])
                continue;

            new_index=index&~flag[i];
            tmp=sol(new_index,player*-1);

            maxx=max(maxx,w[pre[i]][i]+tmp);    //A玩家 
            minx=min(minx,w[pre[i]][i]+tmp);    //B玩家 
        }
    }

    if(player==1)                           //A玩家 
    {
        if(maxx==-INF)                      //找不到可以砍的边 
            maxx=sol(index,-player);
        dp[index][(player+1)/2]=maxx;
    }
    if(player==-1)                          //B玩家 
    {
        if(minx==INF)                       //找不到可以砍的边 
            minx=sol(index,-player);
        dp[index][(player+1)/2]=minx;
    }

    return dp[index][(player+1)/2];
}

int main()
{
    int index;
    while(~scanf("%d",&n))
    {
        n++;
        memset(w,0,sizeof(w));
        memset(c,0,sizeof(c));
        memset(v,0,sizeof(v));

        int x,y,cc,ww;
        for(int i=1;i<n;i++)
        {
            scanf("%d %d %d %d",&x,&y,&cc,&ww);
            c[x][y]=c[y][x]=cc;
            w[x][y]=w[y][x]=ww*cc;          //红边取正,黑边取负 
        }

        pre[0]=-1,dfs(0);                   //预处理 
        flag[0]=0;

        memset(dp,INF,sizeof(dp));

        int ans=sol((1<<(n-1))-1,1);
        printf("%d\n",ans);
    }

    return 0;
}

总结:
挺典型的记忆化搜索的题目,而这道题最大的亮点是可以用二进制来优化时间复杂度。果然太久没做题了有点生疏了,想起今年的校赛,确实很可惜,明年一定要打进省赛!

sicily 1219(记忆化搜索)

标签:

原文地址:http://blog.csdn.net/fuyukai/article/details/46349743

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!