歷屆試題 國王的煩惱（並查集）

#include<bits/stdc++.h>
using namespace std;
const int N=1e4+10;
const int M=1e5+100;
int p[N];
int n,m;
struct P{
	int x,y,time;
}a[M];
bool cmp(P w,P e){
	return w.time>e.time;
}
int find(int u){
	if(p[u]!=u) p[u]=find(p[u]);
	return p[u];
}
bool he(int x,int y){
	x=find(x);
	y=find(y);
	if(x!=y){
		p[x]=y;
		return true;
	}
	return false;
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=0;i<m;i++){
		scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].time);
	}
	for(int i=1;i<=n;i++) p[i]=i;
	sort(a,a+m,cmp);
	int pre=0;
	int ans=0;
	for(int i=0;i<m;i++){
		bool f=he(a[i].x,a[i].y);
		if(f&&a[i].time!=pre){
			ans++;
			pre=a[i].time;
		}
	} 
	printf("%d\n",ans);
	return 0;
}