HDU_5094_dfs

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

bfs,vis[x][y][z]，z表示钥匙的状态，用二进制来表示，key[x][y]储存当前位置钥匙的二进制表示。

注意起始点有钥匙的情况。

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

struct point
{
    int x,y,counts,mykey;
}start;

int n,m,p,k,s,flag,door[][][][],key[][],vis[][][],dir[][] = {-,,,-,,,,};

int main()
{
    while(~scanf("%d%d%d",&n,&m,&p))
    {
        memset(door,-,sizeof(door));
        memset(key,,sizeof(key));
        memset(vis,,sizeof(vis));
        scanf("%d",&k);
        int x1,y1,x2,y2,type;
        while(k--)
        {
            scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&type);
            door[x1][y1][x2][y2] = type;
            door[x2][y2][x1][y1] = type;
        }
        scanf("%d",&s);
        int x,y,keytype;
        while(s--)
        {
            scanf("%d%d%d",&x,&y,&keytype);
            key[x][y] |= <<(keytype-);
        }
        queue<point> q;
        start.x = ;
        start.y = ;
        start.counts = ;
        start.mykey = key[][];
        vis[][][start.mykey] = ;
        q.push(start);
        flag = ;
        while(!q.empty())
        {

            point now = q.front();
            q.pop();
            if(now.x == n && now.y == m)
            {
                flag = ;
                printf("%d\n",now.counts);
                break;
            }
            for(int i = ;i < ;i++)
            {
                point temp;
                temp.x = now.x+dir[i][];
                temp.y = now.y+dir[i][];
                if(temp.x <  || temp.x > n || temp.y <  || temp.y > m)    continue;
                if(door[now.x][now.y][temp.x][temp.y] != - &&(now.mykey & <<(door[now.x][now.y][temp.x][temp.y]-)) == )  continue;
                temp.counts = now.counts+;
                temp.mykey = now.mykey | key[temp.x][temp.y];
                if(vis[temp.x][temp.y][temp.mykey]) continue;
                vis[temp.x][temp.y][temp.mykey] = ;
                q.push(temp);
            }
        }
        if(flag)    printf("-1\n");
    }
}