HDU 1879 繼續暢通工程(最小生成樹)

原題地址

繼續暢通工程

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 22479 Accepted Submission(s): 9600


Problem Description
省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通(但不一定有直接的公路相連,只要能間接通過公路可達即可)。現得到城鎮道路統計表,表中列出了任意兩城鎮間修建道路的費用,以及該道路是否已經修通的狀態。現請你編寫程序,計算出全省暢通需要的最低成本。

Input
測試輸入包含若干測試用例。每個測試用例的第1行給出村莊數目N ( 1< N < 100 );隨後的 N(N-1)/2 行對應村莊間道路的成本及修建狀態,每行給4個正整數,分別是兩個村莊的編號(從1編號到N),此兩村莊間道路的成本,以及修建狀態:1表示已建,0表示未建。

當N爲0時輸入結束。

Output
每個測試用例的輸出佔一行,輸出全省暢通需要的最低成本。

Sample Input
3 1 2 1 0 1 3 2 0 2 3 4 0 3 1 2 1 0 1 3 2 0 2 3 4 1 3 1 2 1 0 1 3 2 1 2 3 4 1 0

Sample Output
3 1 0

最小生成樹。不過在記錄邊的時候要做點小處理:把已經修建好的路,它的成本設爲0,沒修好的路,成本就是輸入的成本。然後再排序,最小生成樹

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++)
        {
            int flag,nums;
            scanf("%d%d%d%d",&node[i].a,&node[i].b,&nums,&flag);
            if(flag)
            {
                node[i].val=0;
            }
            else
            {
                node[i].val=nums;
            }
        }
        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;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章