hdu1863-暢通工程-最小生成樹

暢通工程

Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 16888    Accepted Submission(s): 7072

Problem Description

省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通（但不一定有直接的公路相連，只要能間接通過公路可達即可）。經過調查評估，得到的統計表中列出了有可能建設公路的若干條道路的成本。現請你編寫程序，計算出全省暢通需要的最低成本。

 

Input

測試輸入包含若干測試用例。每個測試用例的第1行給出評估的道路條數 N、村莊數目M ( < 100 )；隨後的 N
行對應村莊間道路的成本，每行給出一對正整數，分別是兩個村莊的編號，以及此兩村莊間道路的成本（也是正整數）。爲簡單起見，村莊從1到M編號。當N爲0時，全部輸入結束，相應的結果不要輸出。

 

Output

對每個測試用例，在1行裏輸出全省暢通需要的最低成本。若統計數據不足以保證暢通，則輸出“?”。

 

Sample Input

3 3 1 2 1 1 3 2 2 3 4 1 3 2 3 2 0 100

 

Sample Output

3 ?

 

Source

浙大計算機研究生複試上機考試-2007年

 

Recommend

lcy


思路：最小生成樹用kruskal算法，把每條邊的按權值的大小從小到大排列，然後然後用並查集判斷當前的兩點是否已經聯通，沒有聯通的話就連接起來 ans加上這條邊的權值，否則就繼續查詢下一條邊，因爲已經按權值從小到大排序了所以可以保證最後全部聯通以後的權值是最小的

代碼：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <cstdlib>
using namespace std;
struct E
{
    int u,v,w;
}e[300];
int fa[300],n,m,rank[300];
int cmp(const void *a, const void *b)
{
    return ((E *)a)->w - ((E *)b)->w;
}
int find(int x)
{
    int q = x;
    while (q!=fa[q])
    {
        q = fa[q];
    }
    int w = x;
    while (w!=q)
    {
        x = fa[w];
        fa[w] = q;
        w = x;
    }
    return w;
}
int uni(int x, int y)
{
    if(x!=y)
    {
        rank [x] += rank[y];   //記錄該樹中已經有幾個節點了
        fa[y] = x;
    }
    return rank[x];
}
int nn;
int f()
{
    int cnt = 0;
    qsort(e,m,sizeof(e[0]),cmp);
    int i;
    int ans = 0;
    for(i = 0; i< m; i++)
    {
        int x = find(e[i].u);
        int y = find (e[i].v);
        if(x!=y)
        {
            nn = uni(x,y);
            ans += e[i].w;
        }
    }
    return ans;

}
int main()
{
    int  i;
    while (scanf("%d %d",&n,&m)&&n)
    {
        nn = 0;
        for(i = 1; i<= m; i++)
        {
            fa[i] = i;
            rank[i] = 1;
        }
        memset(e,0,sizeof(e));
        for(i = 0;i<n;i++)
        {
            scanf("%d %d %d",&e[i].u,&e[i].v,&e[i].w);
        }
        int a = f();
        if(nn < m)
        {
            printf("?\n");
        }
        else
        {
            cout<<a<<endl;
        }
    }
    return 0;
}