hdu_1233 還是暢通工程

原題鏈接：http://acm.hdu.edu.cn/showproblem.php?pid=1233

原題描述：

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算法求最小生成樹，這裏用幷查集查看選中的邊是否構成環。

我的代碼：

#include<stdio.h>
#include<algorithm>
using namespace std;
const int MAXN=5000;
struct Node
{
    int a,b,v;
    bool operator < (const Node& x) const
    {
        return v<x.v;
    }
}p[MAXN];
int f[120]; //用並查集判斷點是否構成了環。
int r[120];
void init()
{
    for(int i=1;i<120;i++)
    {
        f[i]=i;
        r[i]=1;
    }
}
int find(int x)
{
    if(x==f[x]) return x;
    else  return f[x]=find(f[x]);
}
void Union(int a,int b)
{
    a=find(a);
    b=find(b);
    if(a==b) return ;
    else{
        if(r[a]<r[b])
        {
            f[a]=b;
            r[b]+=r[a];
        }
        else
        {
            f[b]=a;
            r[a]+=r[b];
        }
    }
}
int main()
{
    int n;
    while(~scanf("%d",&n)&&n)
    {
        int N=n*(n-1)/2;
        for(int i=0;i<N;i++)
        {
            scanf("%d%d%d",&p[i].a,&p[i].b,&p[i].v);
        }
        sort(p,p+N);
        int ans=0;
        init();
        for(int i=0;i<N;i++)
        {
            int a=find(p[i].a);
            int b=find(p[i].b);
            if(a!=b)
            {
                ans+=p[i].v;
                Union(a,b);
            }
        }
        printf("%d\n",ans);

    }


    return 0;
}

總結：突然在想，好像直接用數組標記點用過沒有，來判斷是否構成環會更簡單吧-，-,今天發現用數組這樣標記是不行的，是否用過和構成環不是一樣的..