牛客国庆集训派对Day3 - I Metropolis （最短路）

https://www.nowcoder.com/acm/contest/203/I

 

POINT：

把大都会都先放入队列。相当于多起点的最短路。

然后进行最短路。记录每个点离他最近的大都会（来源）。这个在松弛的时候就可以记下来。

然后对于每一条边，两个端点的大都会不同时，可以知道这两个大都会的距离为dis[u]+dis[v]+w.

更新答案。

 

#include <vector>
#include <stdio.h>
#include <iostream>
#include <stdio.h>
#include <queue>
using namespace std;
#define LL long long
const int N = 2e5+6;
const LL inf = 0x3f3f3f3f3f3f3f3f;
typedef pair<LL,int> pii;

vector<int>G[N];
vector<int>W[N];
int x[N],from[N],vis[N];
LL dis[N],ans[N];
int n,m,p;
void dij()
{
	priority_queue<pii,vector<pii>,greater<pii> > q;
	for(int i=1;i<=n;i++){
		dis[i]=inf;vis[i]=0;ans[i]=inf;
	}
	for(int i=1;i<=p;i++){
		q.push(make_pair(dis[x[i]]=0, x[i]));
		from[x[i]]=x[i];
	}
	while(!q.empty()){
		pii now=q.top();
		q.pop();
		int u = now.second;
		if(vis[u]) continue;
		vis[u]=1;
		for(int i=0;i<G[u].size();i++){
			int v=G[u][i];
			int w=W[u][i];
			if(dis[v]>dis[u]+w){
				dis[v]=dis[u]+w;
				from[v]=from[u];
				q.push(make_pair(dis[v], v));
			}else if(from[v]!=from[u]){
				int x=from[v],y=from[u];
				ans[x]=min(ans[x],dis[u]+dis[v]+w);
				ans[y]=min(ans[y],dis[u]+dis[v]+w);
			}
		}
	}
}

int main()
{
	scanf("%d%d%d",&n,&m,&p);
	for(int i=1;i<=p;i++)
		scanf("%d",&x[i]);
	for(int i=1;i<=m;i++){
		int a,b,l;
		scanf("%d%d%d",&a,&b,&l);
		G[a].push_back(b);G[b].push_back(a);
		W[a].push_back(l);W[b].push_back(l);
	}
	dij();
	for(int i=1;i<=p;i++){
		printf("%lld%c",ans[x[i]],i==p?'\n':' ');
	}

}