HDU 1233 還是暢通工程（最小生成樹kruskal）

原題地址

還是暢通工程

Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)

Total Submission(s): 41231 Accepted Submission(s): 18824

Problem Description

某省調查鄉村交通狀況，得到的統計表中列出了任意兩村莊間的距離。省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通（但不一定有直接的公路相連，只要能間接通過公路可達即可），並要求鋪設的公路總長度爲最小。請計算最小的公路總長度。

Input

測試輸入包含若干測試用例。每個測試用例的第1行給出村莊數目N ( < 100 )；隨後的N(N-1)/2行對應村莊間的距離，每行給出一對正整數，分別是兩個村莊的編號，以及此兩村莊間的距離。爲簡單起見，村莊從1到N編號。
當N爲0時，輸入結束，該用例不被處理。

Output

對每個測試用例，在1行裏輸出最小的公路總長度。

Sample Input

3 1 2 1 1 3 2 2 3 4 4 1 2 1 1 3 4 1 4 1 2 3 3 2 4 2 3 4 5 0

Sample Output

3 5

裸的最小生成樹。用kruskal算法，結構體存邊，排序，加上並查集判斷是否成環。

判斷成環就是判斷線的兩個點是否有同一個祖先，若有則成環，沒有則可以添加。

AC代碼：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

struct nodes//存邊
{
    int a;
    int b;
    int val;
}node[11111];
bool cmp(nodes a,nodes b)
{
    return a.val<b.val;
}
int par[111];

int fin(int a)//並查集查詢
{
    int p=a;
    while(par[p]!=p)
    {
        p=par[p];
    }
    int x=a,y;
    while(x!=par[x])
    {
        y=par[x];
        par[x]=p;
        x=y;
    }
    return p;
}
void join(int a,int b)//並查集添加邊
{
    int p1=fin(a);
    int p2=fin(b);
    if(p1!=p2)
    {
        par[p1]=p2;
    }
}
int main()
{
    int n;
    while(~scanf("%d",&n)&&n)
    {
        for(int i=0;i<n;i++)
        {
            par[i+1]=i+1;
        }
        int m=(n-1)*n/2;
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&node[i].a,&node[i].b,&node[i].val);
        }
        sort(node,node+m,cmp);

        int ans=0;
        int k=0;
        for(int i=0;i<m;i++)
        {
            if(fin(node[i].a)!=fin(node[i].b))//如果不成環則添加
            {
                ans+=node[i].val;
                join(node[i].a,node[i].b);
                k++;
                if(k==n-1) break;//如果已經添加了n-1條邊則退出
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}