本题出自洛谷p1195,记录一下并查集的经典用法
口袋的天空
1.由于是有权值的一个图,所以可以用并查集来实现。由于要算出最小的和,所以先进行一次排序,将权值最小的边放在队首。由于要分成K份,总数为N,所以用一个计数器(N–)。如果N的大小恰好等于K时,说明已经分成了K份,剩下的没在并查集里的都是权值大的。
2.代码
#include<algorithm>
#include<cstdio>
#include<iostream>
#include<queue>
using namespace std;
inline int read()
{
int p,data=0;
char ch=0;
while((ch!='-')&&ch<'0'||ch>'9') ch=getchar();
if(ch=='-')
{
p=-1;
ch=getchar();
}
else
{
p=1;
}
while(ch>='0'&&ch<='9')
{
data=data*10+ch-'0';
ch=getchar();
}
return data*p;
}
int pre[1100];
//int map[1100][1100];
struct value
{
int j,k,l;
}val[11000];
int n,m,k;
int a,b,c;
int cmp(value x,value y)
{
return x.l<y.l;
}
int tot=0;
int find(int q)
{
if(pre[q]==q) return q;
pre[q]=find(pre[q]);//优化时间复杂度
return pre[q];
}//一个搜索操作,用来找出祖父是谁。
int main()
{
n=read();
m=read();
k=read();
for(int i=1;i<=n;i++)
{
pre[i]=i;//并查集の初始化。
}
for(int i=1;i<=m;i++)
{
a=read();
b=read();
c=read();
val[i].j=a;
val[i].k=b;
val[i].l=c;//快读233,第一次用。
}
sort(val+1,val+m+1,cmp);//将权值进行排序。
for(int i=1;i<=m;i++)
{
if(find(val[i].j)!=find(val[i].k))//如果祖父不是同一个,那么将其加入同一个集,
{
pre[find(val[i].j)]=find(val[i].k);//使他们的祖父成为同一个
n--;//萌萌的充当计数君
tot+=val[i].l;
}
if(n==k)
{
cout<<tot;
return 0;
}
}
cout<<"No Answer";//如果无法成为K组,输出NO ANSWER
return 0;
}
/
任何一个伟大的目标,都有一个微不足道的开始!
——by 洛谷 新手村