hdu 1885(状态压缩+bfs)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1885

思路：对于钥匙，才4把，直接状态压缩搞一下就好了，然后就是开个三位数组标记状态了，跟普通bfs没什么区别。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
#define MAXN 110
struct Node{
    int x,y,step;
    int key;
};
char map[MAXN][MAXN];
bool mark[MAXN][MAXN][22];
int dir[4][2]={{-1,0},{1,0},{0,-1},{0,1}};
int n,m;
Node st;
char Door[4]={'B','Y','R','G'};
char Key[4]={'b','y','r','g'};

void bfs(){
    memset(mark,false,sizeof(mark));
    queue<Node>Q;
    Node p,q;
    st.key=st.step=0;
    mark[st.x][st.y][st.key]=true;
    Q.push(st);
    while(!Q.empty()){
        p=Q.front();
        Q.pop();
        if(map[p.x][p.y]=='X'){
            printf("Escape possible in %d steps.\n",p.step);
            return ;
        }
        for(int i=0;i<4;i++){
            q.x=p.x+dir[i][0];
            q.y=p.y+dir[i][1];
            q.step=p.step+1;
            q.key=p.key;
            if(q.x<1||q.x>n||q.y<1||q.y>m||map[q.x][q.y]=='#')
                continue;
            if(isupper(map[q.x][q.y])&&map[q.x][q.y]!='X'){
                for(int j=0;j<4;j++){
                    if(map[q.x][q.y]==Door[j]){
                        if(q.key&(1<<j)){
                            if(!mark[q.x][q.y][q.key]){
                                mark[q.x][q.y][q.key]=true;
                                Q.push(q);
                            }
                            break;
                        }
                    }
                }
            }else if(islower(map[q.x][q.y])){
                for(int j=0;j<4;j++){
                    if(map[q.x][q.y]==Key[j]){
                        if((q.key&(1<<j))==0){
                            q.key+=(1<<j);
                        }
                        if(!mark[q.x][q.y][q.key]){
                            mark[q.x][q.y][q.key]=true;
                            Q.push(q);
                        }
                    }
                }
            }else {
                if(!mark[q.x][q.y][q.key]){
                    mark[q.x][q.y][q.key]=true;
                    Q.push(q);
                }
            }
        }
    }
    puts("The poor student is trapped!");
}


int main(){
    while(scanf("%d%d",&n,&m),(n+m)){
        for(int i=1;i<=n;i++){
            scanf("%s",map[i]+1);
            for(int j=1;j<=m;j++){
                if(map[i][j]=='*')st.x=i,st.y=j;
            }
        }
        bfs();
    }
    return 0;
}