hdu~1885(bfs+状态压缩)

Key Task

一开始用dfs果断超时，贴上超时代码：

<span style="font-size:18px;">#include <stdio.h>
#include <iostream>
#include <string.h>
#define MAX 0xffffff

int m,n,flag,tt,sx,sy;
char key[]={'y','g','r','b'};
char door[]={'Y','G','R','B'};
int dirx[]={0,1,0,-1},diry[]={1,0,-1,0};
char maze[105][105];
bool mark[105][105][22];

void dfs(int x,int y,int t,int k)
{
    if(flag)
        return ;
    if(maze[x][y]=='X')
    {
        tt=t;
        flag=1;
        return ;
    }
    if(x<0 || x>=n || y<0 || y>=m ||  maze[x][y]=='#' || mark[x][y][k])
        return ;
    if(isupper(maze[x][y]) && maze[x][y]!='X')     //це
    {
        for(int i=0;i<=4;i++)
            if(maze[x][y]==door[i])
            {
                if((k&(1<<i))==0)
                    return ;
            }
    }
    else if(islower(maze[x][y]))    //т©Ёв
    {
        for(int i=0;i<=4;i++)
            if(maze[x][y]==key[i])
            {
                k|=(1<<i);
                break;
            }
    }
    for(int i=0;i<4;i++)
    {
        int fx=x+dirx[i];
        int fy=y+diry[i];
        mark[x][y][k]=true;
        dfs(fx,fy,t+1,k);
        mark[x][y][k]=false;
    }
}

int main()
{
    while(scanf("%d %d",&n,&m) &&(m || n))
    {
        memset(mark,0,sizeof(mark));
        flag=0;
        tt=-1;
        for(int i=0;i<n;i++)
        {
            scanf("%s",maze[i]);
            for(int j=0;j<m;j++)
                if(maze[i][j]=='*')
                    sx=i,sy=j;
        }
        dfs(sx,sy,0,0);
        if(tt!=-1)
            printf("Escape possible in %d steps.\n",tt);
        else
            printf("The poor student is trapped!\n");
    }
    return 0;
}
</span>

bfs+状态压缩

这题和普通的迷宫不一样的是多了4种钥匙以及对应的门

开一个三维数组mark[x][y][k]用来存各个位置的各种状态，通俗的说就是，判断在"k"这种状态下，x，y这个位置是否走过。

用二进制存得到钥匙的状态，用4位即可。

上代码：

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <queue>

using namespace std;
typedef struct ac
{
    int x,y,t;
    int k;          //用二进制存4把钥匙
}node;

int dirx[]={0,1,0,-1},diry[]={1,0,-1,0};    
bool mark[105][105][20];            //判断在有不同的key的各个位置是否出现过
int m,n;
char key[]={'b','g','y','r'};
char door[]={'B','G','Y','R'};
char maze[105][105];    

void bfs(node s)
{
    queue <node> q;
    mark[s.x][s.y][s.k]=true;
    q.push(s);
    while(!q.empty())
    {
        node r=q.front();
        q.pop();
        if(maze[r.x][r.y]=='X')
        {
            printf("Escape possible in %d steps.\n",r.t);
            return ;
        }
        for(int i=0;i<4;i++)
        {
            node w=r;
            w.x+=dirx[i];
            w.y+=diry[i];
            w.t++;
            if(w.x<0 || w.x>=n || w.y<0 || w.y>=m || maze[w.x][w.y]=='#')//撞墙
                continue ;
            if(isupper(maze[w.x][w.y]) && maze[w.x][w.y]!='X')  //door
            {
                for(int i=0;i<4;i++)
                {          //找到对应的门，且有该门的钥匙，且在这种状态下的这个位置未走过
                    if(maze[w.x][w.y]==door[i] && (w.k & (1<<i)) && (!mark[w.x][w.y][w.k]))
                    {
                        mark[w.x][w.y][w.k]=true;
                        q.push(w);
                    }

                }
            }
            else if(islower(maze[w.x][w.y]))    //key
            {
                for(int i=0;i<4;i++)
                {
                    if(maze[w.x][w.y]==key[i])
                    {
                        w.k|=(1<<i);    //key记录得到这把钥匙
                        if(!mark[w.x][w.y][w.k])
                        {
                            mark[w.x][w.y][w.k]=true;
                            q.push(w);
                        }
                    }
                }
            }
            else if(!mark[w.x][w.y][w.k])   //普通的路
            {
                mark[w.x][w.y][w.k]=true;
                q.push(w);
            }
        }
    }
    printf("The poor student is trapped!\n");
}
int main()
{
    while(scanf("%d %d",&n,&m) && (m ||n))
    {
        node s;
        memset(mark,0,sizeof(mark));
        for(int i=0;i<n;i++)
        {
            scanf("%s",maze[i]);
            for(int j=0;j<m;j++)
                if(maze[i][j]=='*')
                    s.x=i,s.y=j;
        }
        s.k=0;
        s.t=0;
        bfs(s);
    }
    return 0;
}