【2020.10.29 牛客 普及组 模拟赛6】T4 自由世界

题目描述

牛牛最近在玩某款游戏，其地图不能看成一个平面直角坐标系，而类似于一张无向图。

地图上存在n个小镇，小镇从1到n编号。有m条道路连接两个小镇，每条道路有其长度$w_i$。

牛牛在k个小镇建立了传送门，也就是说，牛牛可以在任何时候任何瞬间不花费任何代价，直接到达这k个小镇的任何一个。

牛牛一开始在小镇1，牛牛想按1到n的顺序访问所有小镇按顺序做任务，问牛牛需要走过的最短长度是多少。

牛牛可以提前到达某个小镇，但是必须做完前一个小镇的任务，才能做下一个小镇的任务。做任务本身不会增加长度。

---

输入描述:

第一行输入三个整数$n,m,k$，表示地图上小镇的数目，连接小镇道路的条数和建立了传送门的小镇的个数。

随后m行，其中第i行输入三个整数$u_i,v_i,w_i，$表示有一条道路连接了小镇$u_i$ 和小镇$v_i$ ，长度为$w_i$。

随后一行输入k个整数，表示建立了传送门的小镇编号。

对于$10$%的数据有$n\le 3,k=0$。

对于$30$%的数据有$k=0$。

另有$30$%的数据有$m=n-1$。

对于$60$%的数据有$n\le 300$。

对于$100$%的数据有$1\le n\le 2000,n-1\le m\le 5000,0\le k\le n,1\le u_i,v_i\le n,1\le w_i\le 1000$。

数据保证给定的小镇两两相互可达。

注意，连接某两个小镇的可能有多条道路，也有可能有道路的两端是同一个小镇。

输出描述:

一行一个整数表示答案。

---

示例1

输入
5 6 1
1 4 9
1 3 66
3 2 27
4 2 10
2 5 62
3 5 92
3

输出

128

---

说明
牛牛一开始在小镇1，完成任务后，步行至小镇4，随后步行至小镇2，共计长度$9+10=19$。
完成任务后，传送至小镇3。
从小镇3完成任务后，步行至小镇2，再步行至小镇4，这里长度为$27+10=37$，共计长度为$37+19=56$。
然后从小镇4完成任务后步行至小镇2， 然后步行至小镇5，这里长度为$10+62=72$，共计长度为$56+72=128$。
共计长度为128。

---

解题思路

---

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#include<queue>
using namespace std;
int n,m,kk,xx[5050];
int ans,t,h,tt,x,y,w,head[100050],v[5050],dis[5050],k[5050],q[100050];
struct c{
	int x,next;
	long long w;
}a[100050];
void add(int x,int y,int w){
	++tt;
	a[tt].x=y;
	a[tt].next=head[x];
	a[tt].w=w;
	head[x]=tt;
}
void bfs(int s)
{
	int x;
	memset(v,0,sizeof(v));
	memset(dis,0x7f,sizeof(dis));
	memset(q,0,sizeof(q));
	v[s]=1;
	q[1]=s;
	dis[s]=0;
	h=0;
	t=1;
	for(int i = 1; i <= kk; i++){
        dis[xx[i]] = 0;
        v[xx[i]] = 1;
        q[++t]=xx[i];
    }
	do{
		h++;
		x=q[h];
		for(int i=head[x];i;i=a[i].next)
		{
			if(dis[x]+a[i].w<dis[a[i].x])
			{
				dis[a[i].x]=dis[x]+a[i].w;
				if(!v[a[i].x])
				{
					v[a[i].x]=1;
					t++;
					q[t]=a[i].x;
				}
			}
		} 
		v[x]=0;
	}while(h<t);
}
int main() {
	scanf("%d%d%d",&n,&m,&kk);
	for(int i=1; i<=m; i++) 
	{
		scanf("%d%d%d",&x,&y,&w);
		add(x,y,w);
		add(y,x,w);
	}
	for(int i=1; i<=kk; i++) 
	{
		scanf("%d",&xx[i]);
		k[xx[i]]++;
	}
	for(int i=1;i<n;i++)
	{
		if(!k[i+1])
		{
			bfs(i);
			ans=ans+dis[i+1];
		} 
	}
	printf("%d",ans);
}