AcWing 1131. 拯救大兵瑞恩

题目描述：
1944 年，特种兵麦克接到国防部的命令，要求立即赶赴太平洋上的一个孤岛，营救被敌军俘虏的大兵瑞恩。

瑞恩被关押在一个迷宫里，迷宫地形复杂，但幸好麦克得到了迷宫的地形图。

迷宫的外形是一个长方形，其南北方向被划分为 N 行，东西方向被划分为 M 列， 于是整个迷宫被划分为 N×M 个单元。

每一个单元的位置可用一个有序数对 (单元的行号, 单元的列号) 来表示。

南北或东西方向相邻的 2 个单元之间可能互通，也可能有一扇锁着的门，或者是一堵不可逾越的墙。

注意： 门可以从两个方向穿过，即可以看成一条无向边。

迷宫中有一些单元存放着钥匙，同一个单元可能存放 多把钥匙，并且所有的门被分成 P 类，打开同一类的门的钥匙相同，不同类门的钥匙不同。

大兵瑞恩被关押在迷宫的东南角，即 (N,M) 单元里，并已经昏迷。

迷宫只有一个入口，在西北角。

也就是说，麦克可以直接进入 (1,1) 单元。

另外，麦克从一个单元移动到另一个相邻单元的时间为 1，拿取所在单元的钥匙的时间以及用钥匙开门的时间可忽略不计。

试设计一个算法，帮助麦克以最快的方式到达瑞恩所在单元，营救大兵瑞恩。

输入格式
第一行有三个整数,分别表示 N,M,P 的值。

第二行是一个整数 k，表示迷宫中门和墙的总数。

接下来 k 行，每行包含五个整数，Xi1,Yi1,Xi2,Yi2,Gi：当 Gi≥1 时，表示 (Xi1,Yi1) 单元与 (Xi2,Yi2) 单元之间有一扇第 Gi 类的门，当 Gi=0 时，表示 (Xi1,Yi1) 单元与 (Xi2,Yi2) 单元之间有一面不可逾越的墙。

接下来一行，包含一个整数 S，表示迷宫中存放的钥匙的总数。

接下来 S 行，每行包含三个整数 Xi1,Yi1,Qi，表示 (Xi1,Yi1) 单元里存在一个能开启第 Qi 类门的钥匙。

输出格式
输出麦克营救到大兵瑞恩的最短时间。

如果问题无解，则输出 -1。

数据范围
|Xi1−Xi2|+|Yi1−Yi2|=1,
0≤Gi≤P,
1≤Qi≤P,
1≤N,M,P≤10,
1≤k≤150
输入样例：

4 4 9
9
1 2 1 3 2
1 2 2 2 0
2 1 2 2 0
2 1 3 1 0 
2 3 3 3 0
2 4 3 4 1
3 2 3 3 0
3 3 4 3 0
4 3 4 4 0
2
2 1 2 
4 2 1

输出样例：

14

思路：
比较有意思的是state的应用
使用二进制来记录当前状态，比方说第i个状态就可以用$key[g[x][y]]|=1<<i$表示增加i位置的钥匙
检测当前位置是否有位置可以用 $1$ << $w[i]$ &$state$
若当前有钥匙，若新状态的dist大于原状态那么需要进行更新，并入队。由于捡钥匙并不消耗时间，所以是push_front
之后去检测临边，若邻边的dist大于原状态dist+1，那么需要更新，并入队。但是有一个特殊情况，就是如果有门但是没钥匙，就需要continue
代码：

#include<iostream>
#include<cstring>
#include<set>
#include<queue>
using namespace std;
typedef pair<int,int>PII;
set<PII>edges;
const int N=1e3;
int g[N][N];
int e[N],ne[N],h[N],w[N],idx=0;
int dist[N][1<<12];
bool st[N][N];
int key[N];
int n,m,p;
int dx[]={1,-1,0,0},dy[]={0,0,1,-1};
void add(int a,int b,int c)
{
    w[idx]=c;
    e[idx]=b;
    ne[idx]=h[a];
    h[a]=idx++;
}
int bfs()
{
    dist[1][0]=0;
    deque<PII>q;
    q.push_back({1,0});
    while(!q.empty())
    {
        PII t=q.front();
        q.pop_front();
        int pos=t.first;
        int state=t.second;
        if(st[pos][state])
        {
            continue;
        }
        st[pos][state]=true;
        if (pos == n * m) return dist[pos][state];
        if(key[pos])
        {
            int new_state=state|key[pos];
            if(dist[pos][new_state]>dist[pos][state] )
            {
                dist[pos][new_state]=dist[pos][state] ;
                q.push_front({pos,new_state});
            }
        }
        for(int i=h[pos];i!=-1;i=ne[i])
        {
            int j=e[i];
            if((1<<w[i] & state)==0 && w[i])
            {
                continue; 
            }
            if(dist[j][state]>dist[pos][state]+1)
            {
                dist[j][state]=dist[pos][state]+1;
                q.push_back({j, state});
            }
        }
    }
    return -1;
}
int main()
{
    memset(dist, 0x3f, sizeof dist);
    memset(h,-1,sizeof h);
    cin>>n>>m>>p;
    int t=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            g[i][j]=++t;
        }
    }
    int k;
    cin>>k;
    for(int i=0;i<k;i++)
    {
        int x1,y1,x2,y2,c;
        cin>>x1>>y1>>x2>>y2>>c;
        int a=g[x1][y1];
        int b=g[x2][y2];
        edges.insert({a,b});
        edges.insert({b,a});
        if(c)
        {
            add(a,b,c);
            add(b,a,c);
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            for(int u=0;u<4;u++)
            {
                int a=i+dx[u];
                int b=j+dy[u];
                if(!(a>=1 && a<=n && b>=1 && b<=m))
                {
                    continue;
                }
                int x=g[i][j];
                int y=g[a][b];
                if(!edges.count({x,y}))
                {
                    add(x,y,0);
                    add(y,x,0);
                }
            }
        }
    }
    
    int s;
    cin >> s;
    while (s -- )
    {
        int x, y, c;
        cin >> x >> y >> c;
        key[g[x][y]] |= 1 << c;
    }
    
    cout << bfs() << endl;

    return 0;
}