2475: 拯救大兵瑞恩★★★★

题目描述

1944 年，特种兵麦克接到国防部的命令，要求立即赶赴太平洋上的一个孤岛，营救被敌军俘虏的大兵瑞恩。瑞恩被关押在一个迷宫里，迷宫地形复杂，但幸好麦克得到了迷宫的地形图。迷宫的外形是一个长方形，其南北方向被划分为n 行，东西方向被划分为m列， 于是整个迷宫被划分为n*m个单元。每一个单元的位置可用一个有序数对 (单元的行号, 单元的列号) 来表示。南北或东西方向相邻的2个单元之间可能互通，也可能有一扇锁着的门，或者是一堵不可逾越的墙。迷宫中有一些单元存放着钥匙，并且所有的门被分成p  类， 打开同一类的门的钥匙相同，不同类门的钥匙不同。

大兵瑞恩被关押在迷宫的东南角，即(n,m)单元里，并已经昏迷。迷宫只有一个入口， 在西北角。也就是说，麦克可以直接进入(1,1)单元。另外，麦克从一个单元移动到另一个 相邻单元的时间为1，拿取所在单元的钥匙的时间以及用钥匙开门的时间可忽略不计。

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

 输入

 输出

输出麦克营救到大兵瑞恩的最短时间。如果问题无解，则输出-1

代码（本人特色，不写批注（其实是懒））

#pragma GCC optimize(2)

#include<bits/stdc++.h>

     

using namespace std;

typedef long long ll;

#define SIS std::ios::sync_with_stdio(false)

#define space putchar(' ')

#define enter putchar('\n')

#define lson root<<1

#define rson root<<1|1

typedef pair<int,int> PII;

typedef pair<int,PII> PIII;

const int mod=1e9+7;

const int N=2e5+5;

const int inf=0x7f7f7f7f;

    

int gcd(int a,int b)

{

    return b==0?a:gcd(b,a%b);

}

     

ll lcm(ll a,ll b)

{

    return a*(b/gcd(a,b));

}

     

template <class T>

void read(T &x)

{

    char c;

    bool op = 0;

    while(c = getchar(), c < '0' || c > '9')

        if(c == '-')

            op = 1;

    x = c - '0';

    while(c = getchar(), c >= '0' && c <= '9')

        x = x * 10 + c - '0';

    if(op)

        x = -x;

}

template <class T>

void write(T x)

{

    if(x < 0)

        x = -x, putchar('-');

    if(x >= 10)

         write(x / 10);

    putchar('0' + x % 10);

}

ll qsm(int a,int b,int p)

{

    ll res=1%p;

    while(b)

    {

        if(b&1)

            res=res*a%p;

        a=1ll*a*a%p;

        b>>=1;

    }

    return res;

}

struct node

{

    int to,nex,w;

}edge[400];

const int P=1<<10;

int head[400],dis[400][P],vis[400][P];

int g[400][400],key[400];

int tot;

int n,m,s,k;

set<PII> edges;

void add(int u,int v,int w)

{

    edge[tot].w=w;

    edge[tot].to=v;

    edge[tot].nex=head[u];

    head[u]=tot++;

}

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

    

void build()

{

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

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

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

      {

          int x=i+dir[u][0],y=j+dir[u][1];

          if(!x||x>n||!y||y>m)continue;

          int a=g[i][j],b=g[x][y];

          if(edges.count({a,b})==0){

              add(a,b,0);

              add(b,a,0);

          }

      }

       

}

    

int bfs()

{

    memset(dis,inf,sizeof dis);

    memset(vis,0,sizeof vis);

    dis[1][0]=0;

    deque<PII> q;

    q.push_back({1,0});

    while(q.size())

    {

            

        PII now=q.front();q.pop_front();

        int u=now.first,sta=now.second;

        //cout<<u<<endl;

        if(vis[u][sta])continue;

        vis[u][sta]=1;

        if(u==n*m) return dis[u][sta];

            

        if(key[u])

        {

            int state=sta|key[u];

            if(dis[u][state]>dis[u][sta])

            {

                dis[u][state]=dis[u][sta];

                q.push_front({u,state});

            }

        }

       // cout<<"2--"<<u<<endl;

        for(int i=head[u];~i;i=edge[i].nex)

        {

            int v=edge[i].to,w=edge[i].w;

            if(w && !(sta >> w - 1 & 1) ) continue;

            if(dis[v][sta]>dis[u][sta]+1)

            {

                dis[v][sta]=dis[u][sta]+1;

                q.push_back({v,sta});

            }

        }

    

    }

    return -1;

    

}

    

    

    

    

int main()

{

   cin>>n>>m>>s>>k;

   memset(head,-1,sizeof head);

       

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

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

           g[i][j]=t++;

    

   while(k--)

   {

       int x1,x2,y1,y2,op;

       cin>>x1>>y1>>x2>>y2>>op;

       int a=g[x1][y1],b=g[x2][y2];

       edges.insert({a,b});edges.insert({b,a});

       if(op){

           add(a,b,op);add(b,a,op);

       }

    

   }

    build();

    

   int rz;

   cin>>rz;

   while(rz--)

   {

       int x,y,c;

       cin>>x>>y>>c;

       key[g[x][y]]|=1<<c-1;

          

    

   }

      

   cout<<bfs()<<endl;

    

    

   return 0;

    

}