BFS+状态压缩 hdu-1885-Key Task

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=1885

题目意思：

给一个矩阵，给一个起点多个终点，有些点有墙不能通过，有些点的位置有门，需要拿到相应颜色的钥匙才能打开，问到达终点的最短步数。

解题思路：

BFS+状态压缩。

将每种颜色对应一个二进制数位，1表示已经得到该颜色的钥匙，0表示没有得到。

一把钥匙可以同种颜色的多扇门。

代码：

 

#include<iostream>

#include<cmath>

#include<cstdio>

#include<cstdlib>

#include<string>

#include<cstring>

#include<algorithm>

#include<vector>

#include<map>

#include<set>

#include<stack>

#include<list>

#include<queue>

#define eps 1e-6

#define INF 0x1f1f1f1f

#define PI acos(-1.0)

#define ll __int64

#define lson l,m,(rt<<1)

#define rson m+1,r,(rt<<1)|1

//#pragma comment(linker, "/STACK:1024000000,1024000000")

using namespace std;



/*

freopen("data.in","r",stdin);

freopen("data.out","w",stdout);

*/



map<char,int>myp1,myp2;



bool vis[20][110][110];

char save[110][110];

int n,m,dir[4][2]={{-1,0},{0,1},{1,0},{0,-1}};



struct Point

{

   int x,y,step,ss;

}s;



bool iscan(int x,int y)

{

   if(x<1||x>n||y<1||y>m||save[x][y]=='#')

      return false;

   return true;

}

//一把钥匙可以开颜色相同的多种锁

void bfs()

{

   memset(vis,false,sizeof(vis));

   s.step=0,s.ss=0;

   queue<Point>myq;



   myq.push(s);

   vis[0][s.x][s.y]=true;

   while(!myq.empty())

   {

      Point tmp=myq.front();

      myq.pop();



      int xx,yy;



      for(int i=0;i<4;i++)

      {

         xx=tmp.x+dir[i][0],yy=tmp.y+dir[i][1];

         if(!iscan(xx,yy))

            continue;

         if(save[xx][yy]=='X')

         {

            printf("Escape possible in %d steps.\n",tmp.step+1);

            return ;

         }



         int tt=tmp.ss;

         if(myp1.find(save[xx][yy])!=myp1.end()) //得到一把钥匙

         {

            tt=tt|(1<<myp1[save[xx][yy]]);

            if(vis[tt][xx][yy]) //该状态之前已被访问

               continue;

         }

         else if(myp2.find(save[xx][yy])!=myp2.end())

         {

            if(!(tt&(1<<myp2[save[xx][yy]]))) //没有钥匙

               continue;

         }

         if(vis[tt][xx][yy])

            continue;

         vis[tt][xx][yy]=true;

         Point t;

         t.x=xx,t.y=yy,t.ss=tt,t.step=tmp.step+1;

         myq.push(t);

      }

   }

   printf("The poor student is trapped!\n");

   return ;

}



int main()

{

   myp1['b']=0,myp1['y']=1,myp1['r']=2,myp1['g']=3; //每种不同的颜色对应不同的数位

   myp2['B']=0,myp2['Y']=1,myp2['R']=2,myp2['G']=3;



   while(scanf("%d%d",&n,&m)&&m+n)

   {

      memset(vis,false,sizeof(vis));

      for(int i=1;i<=n;i++)

      {

         scanf("%s",save[i]+1);

         for(int j=1;j<=m;j++)

         {

            if(save[i][j]=='*')

               s.x=i,s.y=j;

         }

      }

      bfs();

   }



   return 0;

}