hdu5094(BFS&状态压缩)

node里面多开一维，表示拥有钥匙的状态。

标记数组多开一维，vis[i][j][k]表示第i行第j列拥有钥匙的状态。

剩下的就是BFS。

这个题一个格子里面有可能有很多把钥匙，这是一个坑点，wa了2发。

#include<cstdio>#include<map>#include<queue>#include<cstring>using namespace std;const int bit = 1<<12;int n,m,p,k,numk;int bmap[55][55];struct node{    int x;    int y;    int step;    int key;    node(int a,int b,int c,int d):x(a),y(b),step(c),key(d){}};map<pair<int,int>,int> door;int vis[51][51][(1<<12)+5];int sumkey[55][55];int dir[4][2] = {1,0,-1,0,0,1,0,-1};void bfs(){    memset(vis,0,sizeof(vis));    queue<node>q;    q.push(node(1,1,0,1|bmap[1][1]));    vis[1][1][1|bmap[1][1]] = 1;    while(!q.empty())    {        node fr = q.front(); q.pop();        for(int i=0;i<4;i++)        {            int nx = fr.x+dir[i][0];            int ny = fr.y+dir[i][1];            int a = fr.x*m+fr.y;            int b = nx*m+ny;            int kk = door[make_pair(a,b)];            if(nx>n||ny>m||nx<=0||ny<=0||!(fr.key&(1<<kk))||vis[nx][ny][fr.key]) continue;            if(nx==n&&ny==m)            {                printf("%d\n",fr.step+1);                return ;            }            vis[nx][ny][fr.key|bmap[nx][ny]] = 1;            q.push(node(nx,ny,fr.step+1,fr.key|bmap[nx][ny]));        }    }    printf("-1\n");}int main(){    //printf("%d\n",(1<<12)*51*51);    int x1,y1,x2,y2,w;    while(scanf("%d%d%d",&n,&m,&p)!=EOF)    {        door.clear();        memset(bmap,0,sizeof(bmap));        memset(sumkey,0,sizeof(sumkey));        scanf("%d",&k);        for(int i=0;i<k;i++)        {            scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&w);            w++;            door[make_pair(x1*m+y1,x2*m+y2)] = w;            door[make_pair(x2*m+y2,x1*m+y1)] = w;        }        scanf("%d",&numk);        while(numk--)        {            scanf("%d%d%d",&x1,&y1,&w);            w++;            bmap[x1][y1] |= (1<<w);        }        bfs();    }    return 0;}