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