luogu P4943 密室

analysis

首先简化问题,即哈利和罗恩是等价的,也就是说罗恩能走的地方哈利都能够走(可怜的韦斯莱),所以我们可以只考虑哈利

对于哈利(哈利+罗恩)最后完成任务的方式,可能有以下三种情况:

1. 从1到达第一个房间+从1到达第二个房间

2. 从1到达第一个房间然后到达第二个房间

3. 从1到达第二个房间后到达第一个房间

我们只需要把这些方案的最短路算出来做一个比较就可以了

但是注意第一种情况中哈利和罗恩还是有区别,因此不能两遍最短路都经过绿色点(蛇佬腔专用点),于是第一种情况中就包含了两种情况,因此严谨来讲,上面的3中情况实际上是4种

于是对于第一种情况,跑两遍dijkstra

第二种和第三种先跑一个以1为起点的dijkstra,然后在以两个终点为起点分别跑两次最短路即可

code

#include<bits/stdc++.h>
using namespace std;
#define loop(i,start,end) for(register int i=start;i<=end;++i)
#define clean(arry,num) memset(arry,num,sizeof(arry))
#define anti_loop(i,start,end) for(register int i=start;i>=end;--i)
#define ll long long
template<typename T>void read(T &x){
	x=0;char r=getchar();T neg=1;
	while(r>'9'||r<'0'){if(r=='-')neg=-1;r=getchar();}
	while(r>='0'&&r<='9'){x=(x<<1)+(x<<3)+r-'0';r=getchar();}
	x*=neg;
}
const int maxn=5e4+10;
const int maxm=1e5+10;
int n,m,k,head[maxn],cnt=0,t1,t2;
bool slq[maxn];
struct node{
	int e;
	ll w;
	int nxt;
}edge[maxm<<1];
inline void addl(int u,int v,ll w){
	edge[cnt].e=v;
	edge[cnt].w=w;
	edge[cnt].nxt=head[u];
	head[u]=cnt++;
}
struct point{
	int pos;
	ll w;
	point():pos(0),w(0){}
	point(int pos,ll w):pos(pos),w(w){}
	friend bool operator<(point a,point b){
		return a.w>b.w;
	}
};
ll dis[maxn];
priority_queue<point>q;
void dijkstra(int s,bool harry){
	clean(dis,0x3f);
	dis[s]=0;
	q.push(point(s,0));
	while(q.empty()==false){
		int f=q.top().pos;
		q.pop();
		for(int i=head[f];i!=-1;i=edge[i].nxt){
			int v=edge[i].e;
			if(slq[v]&&!harry)continue;
			if(dis[v]>dis[f]+edge[i].w){
				dis[v]=dis[f]+edge[i].w;
				q.push(point(v,dis[v]));
			}
		}
	}
}
int main(){
	#ifndef ONLINE_JUDGE
	freopen("datain.txt","r",stdin);
	#endif
	clean(head,-1);
	clean(slq,false);
	read(n);read(m);read(k);
	loop(i,1,k){
		int id;read(id);
		slq[id]=1;
	}
	loop(i,1,m){
		int ai,bi,ci;
		read(ai);read(bi);read(ci);
		addl(ai,bi,ci);
		addl(bi,ai,ci);
	}
	ll res=LONG_LONG_MAX;
	read(t1);read(t2);
	dijkstra(1,true);
	ll H_t1=dis[t1];
	ll H_t2=dis[t2];
	dijkstra(1,false);
	ll R_t1=dis[t1];
	ll R_t2=dis[t2];
	res=min(min(res,max(H_t1,R_t2)),max(H_t2,R_t1));
	
	dijkstra(t1,true);
	ll H_t1_t2=dis[t2];
	dijkstra(t1,false);
	ll R_t1_t2=dis[t2];
	
	dijkstra(t2,true);
	ll H_t2_t1=dis[t1];
	dijkstra(t2,false);
	ll R_t2_t1=dis[t1];
	
	res=min(res,min(min(H_t2+H_t2_t1,H_t1+H_t1_t2),min(R_t2+R_t2_t1,R_t1+R_t1_t2)));
	
	printf("%lld\n",res);
	return 0;
}