还是畅通工程

还是畅通工程

该题是最短路和并查集的结合。
需要注意sort(a,a+len,cmp);函数，此函数在for循环里，如果i是从1开始的，则a需要+1，，，，，一定不要搞混，一直交一直错。

#include<iostream>
#include<algorithm>
using namespace std;
const int n=1e6+5;
long long s[n],ans=0;
struct node
{
    long long x,y,w;
} a[n];
//初始化
void init_set(int x)
{
    for(int i=1; i<=x; i++)
        s[i]=i;
}
//排序
bool cmp(node a,node b)
{
    return a.w<b.w;
}
//查找
int find_set(int x)
{
    return x == s[x]?x:find_set(s[x]);
}
//合并
void union_set(int x,int y,int w)
{
    x=find_set(x);
    y=find_set(y);
    if(x!=y)
    {
        s[x]=s[y];
        ans=ans+w;
    }
}
//排序
int main()
{
    int N;

    while(cin>>N)
    {
        ans=0;
        if(N==0)
            break;//N为0时结束

        int len=N*(N-1)/2;


        init_set(N);


        for(int i=1; i<=len; i++)
        {
            cin>>a[i].x>>a[i].y>>a[i].w;
        }


        sort(a+1,a+len+1,cmp);


        for(int i=1; i<=len; i++)
        {
            union_set(a[i].x,a[i].y,a[i].w);
        }

        cout<<ans<<endl;
    }
    return 0;
}