2020.7.21 T3最小代价（jz暑假训练day6）

Description

给出一幅由n个点m条边构成的无向带权图。
其中有些点是黑点，其他点是白点。
现在每个白点都要与他距离最近的黑点通过最短路连接（如果有很多个黑点，可以选取其中任意一个），我们想要使得花费的代价最小。请问这个最小代价是多少？
注意：最后选出的边保证每个白点到离它最近的黑点的距离仍然等于原图中的最短距离。

Input

第一行两个整数n，m；
第二行n 个整数，0表示白点，1 表示黑点；
接下来m 行，每行三个整数x，y，z，表示一条连接x和y 点，权值为z 的边。

Output

如果无解，输出impossible；
否则，输出最小代价。

Sample Input

5 7
0 1 0 1 0
1 2 11
1 3 1
1 5 17
2 3 1
3 5 18
4 5 3
2 4 5

Sample Output

5
【样例解释】
选 2、4、6三条边

Data Constraint

对30%的输入数据： 1≤n≤10， 1≤m≤20；
对100%的输入数据：1≤n≤100000，1≤m≤200000，1≤z≤1000000000

正解

首先，我们新建一个源点，连接所有黑点，边权为0，然后跑一遍最短路，即可求出每一个白点到黑点的最短距离。然后，利用贪心的思想。我们扫一遍每一个白点，在扫一遍每一条直接连向它的边。对于一条边，如果是从当前的白点走过去，也就是dis[i]=dis[j]+边权值， 那么我们就要找到一条最短的这样的边，并加入答案，其它长的边全部删掉。如果是从对面走到当前这一个白点，则不做任何操作。

代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
#define N 200007
using namespace std;
int n,m,cnt,num[N],head[N<<1];
long long ans,dis[N];
bool bz[N];
struct node{
	int to,w,nxt;
}e[N<<1];
void add(int u,int v,int w){
	e[++cnt].to=v;
	e[cnt].w=w;
	e[cnt].nxt=head[u];
	head[u]=cnt;
}
void spfa(){
	queue<int>q;
	for(int i=1;i<=n;i++)
		dis[i]=10e17,bz[i]=0;
	bz[0]=1;
	dis[0]=0;
	q.push(0);
	while(!q.empty()){
		int np=q.front();
		q.pop();
		bz[np]=0;
		for(int i=head[np];i;i=e[i].nxt){
			int v=e[i].to,w=e[i].w;
			if(dis[v]>dis[np]+w){
				dis[v]=dis[np]+w;
				if(!bz[v])
					bz[v]=1,q.push(v);
			}
		}
	}
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%d",&num[i]);
		if(num[i]) add(0,i,0);
	}
	for(int i=1;i<=m;i++){
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		if(num[u]&&num[v]) continue;
		add(u,v,w),add(v,u,w);
	}
	spfa();
	for(int i=1;i<=n;i++)
		if(dis[i]==10e17){
			printf("impossible");
			return 0;
		}
	for(int i=1;i<=n;i++){
		if(num[i]) continue;
		int k=0x3f3f3f3f;
		for(int j=head[i];j;j=e[j].nxt){
			int v=e[j].to,w=e[j].w;
			if(dis[i]==dis[v]+w||num[v])
				k=min(k,w);
		}
		ans+=k;
	}
	printf("%lld",ans);
}