拯救大兵瑞恩---建图+状压DP+deque+最短路

题目链接

#include <iostream>
#include <cstring>
#include <algorithm>
#include <set>
#include <deque>
#include <vector>
using namespace std;
#define x first
#define y second
typedef pair<int,int>PII;
const int N = 110,M = 1<<10;
vector<PII>g[N];
set<PII>edges;   //判断边是否已经建到图里了
int n,m,p,k;
int dist[N][M];  // 第一维是坐标，第二维是钥匙状态
bool st[N][M];   //判断是否更新过了，如果更新过就是最小值了，不需要在改变了
int id[12][12];  //将二维映射为一维，方便写
int key[N];      //记录每个位置的钥匙，用二进制表示
void build()
{
    int dx[4]={1,0,-1,0},dy[4]={0,1,0,-1};
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            for(int u=0;u<4;u++)
            {
                int x=i+dx[u],y=j+dy[u];
                if(x<1||x>n||y<1||y>m)continue;
                int a=id[i][j],b=id[x][y];
                // 将没有建到图中的可走边建图
                if(!edges.count({a,b}))
                    g[a].push_back({b,0});
            }
        }
    }
}
int bfs()
{
    
    //双端队列，
    memset(dist,0x3f,sizeof dist);
    deque<PII>q;
    q.push_back({1,0});
    dist[1][0]=0;
    
    while(q.size())
    {
        auto t=q.front();
        q.pop_front();
        if(st[t.x][t.y])continue;
        st[t.x][t.y]=true;
        if(t.x==n*m)return dist[t.x][t.y];
        //捡起这个位置的钥匙
        if(key[t.x])
        {
            int state=t.y|key[t.x];
            if(dist[t.x][state]>dist[t.x][t.y])
            {
                dist[t.x][state]=dist[t.x][t.y];
                q.push_front({t.x,state});
            }
        }
        for(auto [v,w] : g[t.x])
        {
            //如果下个位置需要钥匙，但是上一个点更新过来没有这个钥匙直接跳过
            if(w&&!(t.y>>w-1&1))continue;
            if(dist[v][t.y]>dist[t.x][t.y]+1)
            {
                dist[v][t.y]=dist[t.x][t.y]+1;
                q.push_back({v,t.y});
            }
        }
    }
    
    return -1;
}

signed main()
{
    cin>>n>>m>>p>>k;
    for(int i=1,t=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            id[i][j]=t++;
    for(int i=1;i<=k;i++)
    {
        int x1,y1,x2,y2,c;
        cin>>x1>>y1>>x2>>y2>>c;
        int a=id[x1][y1],b=id[x2][y2];
        edges.insert({a,b});
        edges.insert({b,a});
        if(c)g[a].push_back({b,c}),g[b].push_back({a,c});
    }
    
    build();
    
    int s;
    cin>>s;
    while(s--)
    {
        int x1,y1,c;
        cin>>x1>>y1>>c;
        key[id[x1][y1]]|=1<<c-1;
    }
    cout<<bfs()<<endl;
}