SSL_1764 最小生成树

题意

给出点数为n的一个图，求里面的最小生成树。

思路

这里我用的是并查集的方法，先给边的权值排序，从最短的边开始选，如果连接这条边的两个点不在同一个集合里面我们就把它们放进去，答案加上这条边的权值就可以了。

代码

#include<cstdio>
#include<algorithm>
using namespace std;
int n,ans,m,father[5001];
struct node{
    int l,r,v;
}s[25000001];
int cmp(node x,node y){return x.v<y.v;}
int find(int x)//查找父亲
{
    int t=x,t1;
    while (x!=father[x]) x=father[x];
    while(father[t]!=t)//压缩路径
    {
        t1=father[t];
        father[t]=x;
        t=t1;
    }
    return x;
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
      for (int j=1;j<=n;j++)
      {

          scanf("%d",&s[++m].v);
          if (s[m].v&&j>=i) s[m].l=i,s[m].r=j;
          else m--;//l是第一个点，r是第二个点，v是权值
      }
    sort(s+1,s+m+1,cmp);//排序
    for (int i=1;i<=n;i++)
      father[i]=i;//初始化父亲为自己
    for (int i=1;i<=m;i++)//从最短的边开始
    {
        int f1=find(s[i].l),f2=find(s[i].r);
        if (f1!=f2) ans+=s[i].v,father[f2]=f1;
        //如果它们不在同一个集合里就统计答案，并且合并它们
    }
    printf("%d",ans);
}