题目
用Kruscal算法做最小生成树
思路
将所有的边从小到大排序
从最小的边开始加起,若两端点不在同一个集合(用并查集的方法判断),便可以加。
一直加到所有的点都连接起来为止。得出答案。
代码
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN=100000+1;
const int MAXM=1000000+1;
int N,M,vis[MAXM],u[MAXM],v[MAXM],w[MAXM],fa[MAXN];
void qksort(int x,int y)
{
if(x>=y) return;
int i=x,j=y,key=w[(x+y)/2];
while(i<=j)
{
while(w[i]<key) i++;
while(w[j]>key) j--;
if(i<=j)
{
int t;
t=u[i]; u[i]=u[j]; u[j]=t;
t=v[i]; v[i]=v[j]; v[j]=t;
t=w[i]; w[i]=w[j]; w[j]=t;
i++;
j--;
}
}
qksort(x,j);
qksort(i,y);
}
int find(int x)
{
return fa[x]<0 ? x : fa[x]=find(fa[x]);
}
int kruskal()
{
int ans=0;
memset(fa,-1,sizeof(fa));
memset(vis,0,sizeof(vis));
qksort(1,M);
for(int i=1;i<=M;i++)
{
int rx=find(u[i]);
int ry=find(v[i]);
if(rx!=ry)
{
fa[rx]=ry;
ans=ans+w[i];
vis[i]++;
}
}
return ans;
}
int main()
{
scanf("%d%d",&N,&M);
for(int i=1;i<=M;i++) scanf("%d%d%d",&u[i],&v[i],&w[i]);
printf("%d",kruskal());
return 0;
}