Applese 的毒气炸弹(最小生成树(MST)&Kruskal/**详解*/)

链接：https://ac.nowcoder.com/acm/contest/330/G
来源：牛客网

• 题目描述
  众所周知，Applese 是个很强的选手，它的化学一定很好。
  今天他又AK了一套题觉得很无聊，于是想做个毒气炸弹玩。
  毒气炸弹需要 k 种不同类型元素构成，Applese一共有 n 瓶含有这些元素的试剂。
  已知元素混合遵循 m 条规律，每一条规律都可以用 “x y c” 描述。
  表示将第 x 瓶试剂混入第 y 瓶试剂或者把第 y 瓶试剂混入第 x 瓶试剂，需要消耗 c 的脑力。
  特别地，除了这 m 条规律外，Applese 可以将任意两瓶相同元素的试剂混合，且不需要消耗脑力。
  Applese 想要配出毒气炸弹，就需要使 S 中含有
  1∼k 这 k 种元素。它想知道自己最少花费多少脑力可以把毒气炸弹做出来。
• 输入描述:
  第一行为三个整数 n, m, k 表示 Applese 拥有的试剂的数量，混合规律的数量和所需的元素种类数。
  第二行为 n 个整数
  a1,a2,…,an，分别表示每一瓶试剂的元素类型。
  接下来m行，每行三个整数 x, y, c，含义如题目描述中所述。不保证 x, y的试剂种类不同。
• 输出描述:
  输出一个正整数表示最小的耗费脑力。特别地，如果无法合成出毒气炸弹，输出 “-1”。
• 输入
  6 8 2
  1 1 1 2 2 2
  1 2 1
  2 3 2
  1 3 3
  3 4 6
  4 5 1
  4 6 3
  5 6 2
  1 6 2
• 输出
  2
• 备注:1≤n,k,m≤10⁵,1≤x,y≤n,x≠y,1≤c≤10⁹

  此题的意思就是将每一种类型的试剂都使用一次并且总花费最少.我们可以使用最小生成树算法,每种试剂选择一次最后选到的总花费最少.首先我们把每一个x,y都看成一条边,然后按照花费从小到大的顺序排序,遍历每一个x,y如果它们是同一种类型就不用选择,否则就选择这两种试剂,并且把这两类试剂合并成一类.

• 克鲁斯卡尔(Kruskal)算法过程
    将所有的边按照权值从小到大排序,遍历所有的边,看这两个点是否属于同一个连通分量,如果属于就舍去此边选择下一个较小的边,否则,就输出这条边,并且将这两个点合并成一个连通分量.

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<string>
#include<queue>
#include<vector>
#define pi 3.1415926
#define mod 1000000007
using namespace std;

//typedef pair<int,int> Node;
typedef long long LL;
const int Max_n=100005;
int a[Max_n],f[Max_n];
 
struct Edge{
	LL u,v,w;
}edge[Max_n];

int cmp(Edge a,Edge b){
	return a.w<b.w;
}

int getf(int v){
	if(f[v]==-1) return v;//-1说明v是根 
	else return f[v]=getf(f[v]);//不是根继续向上找根 
}

void Merge(int u,int v){//两类试剂合并成一类 
	int fu=getf(u);
	int fv=getf(v);
	f[fu]=fv; 
}
int main(){   
	int n,m,k;
	scanf("%d%d%d",&n,&m,&k);
	memset(f,-1,sizeof(f));
	//for(int i=1;i<=n;i++)
	//	f[i]=-1;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(int i=1;i<=m;i++)
		scanf("%lld%lld%lld",&edge[i].u,&edge[i].v,&edge[i].w);
	sort(edge+1,edge+m+1,cmp);//对m条边进行排序
	int count=0;//已经选择的边的数目 
	LL ans=0;//注意数据类型
	for(int i=1;i<=m;i++){//对m条边进行选择
		//两种试剂是一种类型 
		if(getf(a[edge[i].u])==getf(a[edge[i].v])) continue; 
		Merge(a[edge[i].u],a[edge[i].v]);//不是一种试剂就合并成生成树
		ans+=edge[i].w;//合并的花费 
		count++;
		if(count==k-1) break;//选了k-1条边就退出 
	} 
	if(count==k-1) printf("%lld\n",ans);
	else printf("-1\n"); 
	return 0;
}