题目描述
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;
}
