题干：
 

众所周知，Applese 是个很强的选手，它的化学一定很好。


今天他又AK了一套题觉得很无聊，于是想做个毒气炸弹玩。

毒气炸弹需要 k 种不同类型元素构成，Applese一共有 n 瓶含有这些元素的试剂。

已知元素混合遵循 m 条规律，每一条规律都可以用 "x y c" 描述。

表示将第 x 瓶试剂混入第 y 瓶试剂或者把第 y 瓶试剂混入第 x 瓶试剂，需要消耗 c 的脑力。
 

特别地，除了这 m 条规律外，Applese 可以将任意两瓶相同元素的试剂混合，且不需要消耗脑力。

 

Applese 想要配出毒气炸弹，就需要使 S 中含有 1∼k1∼k 这 k 种元素。它想知道自己最少花费多少脑力可以把毒气炸弹做出来。

输入描述:

第一行为三个整数 n, m, k 表示 Applese 拥有的试剂的数量，混合规律的数量和所需的元素种类数。
第二行为 n 个整数 a1,a2,…,ana1,a2,…,an，分别表示每一瓶试剂的元素类型。
接下来m行，每行三个整数 x, y, c，含义如题目描述中所述。不保证 x, y的试剂种类不同。

输出描述:

输出一个正整数表示最小的耗费脑力。特别地，如果无法合成出毒气炸弹，输出 "-1"。

示例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≤1051≤n,k,m≤105
1≤x,y≤n,x≠y1≤x,y≤n,x≠y
1≤c≤109

 

解题报告：

看成一张图，就是把同类元素的试剂当作一个点之后，求这个图的最小生成树就行了。注意判不连通的情况。

AC代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<map>
#include<vector>
#include<set>
#include<string>
#include<cmath>
#include<cstring>
#define ll long long
#define pb push_back
#define pm make_pair
#define fi first
#define se second
using namespace std;
const int MAX = 2e5 + 5;
struct Edge {
	int u,v;
	int c;
	Edge(){}
	Edge(int u,int v,int c):u(u),v(v),c(c){}
	bool operator<(const Edge b) const{
		return c < b.c;
	}
} e[MAX];
int tot;
int a[MAX];
int f[MAX];
int getf(int v) {
	return f[v] == v ? v : f[v] = getf(f[v]);
}
void merge(int u,int v) {
	int t1 = getf(u);
	int t2 = getf(v);
	f[t2] = t1;
}

int main()
{
	int n,m,k;
	cin>>n>>m>>k;
	for(int i = 1; i<=k; i++) f[i] = i;
	for(int i = 1; i<=n; i++) cin>>a[i];
	for(int x,y,c,i = 1; i<=m; i++) {
		cin>>x>>y>>c;
		if(a[x] != a[y]) e[++tot] = {a[x],a[y],c};
		
	}
	sort(e+1,e+tot+1);
	ll ans = 0,cnt = 0;
	for(int u,v,i = 1; i<=tot; i++) {
		u = e[i].u;
		v = e[i].v;
		
		if(getf(u) == getf(v)) continue;
		cnt++;
		merge(u,v);
		ans += 1ll*e[i].c;
		if(cnt == k-1) break;
	}
	if(cnt != k-1) puts("-1");
	else printf("%lld\n",ans);
	return 0 ;
 }