畅通工程之最低成本建设问题（并查集）

畅通工程之最低成本建设问题

题目

某地区经过对城镇交通状况的调查，得到现有城镇间快速道路的统计数据，并提出“畅通工程”的目标：使整个地区任何两个城镇间都可以实现快速交通（但不一定有直接的快速道路相连，只要互相间接通过快速路可达即可）。现得到城镇道路统计表，表中列出了有可能建设成快速路的若干条道路的成本，求畅通工程需要的最低成本。

输入格式:

输入的第一行给出城镇数目N (1<N≤1000)和候选道路数目M≤3N；随后的M行，每行给出3个正整数，分别是该条道路直接连通的两个城镇的编号（从1编号到N）以及该道路改建的预算成本。

输出格式:

输出畅通工程需要的最低成本。如果输入数据不足以保证畅通，则输出“Impossible”。

输入样例：

6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3

输出样例：

12
本题可以用并查集解决问题。

#include <iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>

using namespace std;
int s[100000];
struct node{
int x,y,z;
};
struct node p[10000];
bool cmp(struct node a, struct node b) { return a.z<b.z; }
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=0;i<=n;i++) s[i]=i;
    for(int i=0;i<m;i++){
        scanf("%d%d%d",&p[i].x,&p[i].y,&p[i].z);
    }
    sort(p,p+m,cmp);
    int t=1;
    int ans=0;
    for(int i=0;i<m;i++)
    {
        int a,b;
        a=p[i].x;
        b=p[i].y;
        while(s[a]!=a)
        {
            a=s[a];
        }
        while(s[b]!=b)
        {
            b=s[b];
        }
        if(a!=b)
        {
            s[a]=b;
            t++;
            ans+=p[i].z;
        }
        if(t==n)
            break;

    }
    if(t==n)
        printf("%d\n",ans);
    else
    {
        printf("Impossible");
    }
    return 0;
}