hdu 1879 繼續暢通工程————最小生成樹(Kruskal)

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

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

思路:

kruskal模板題,對於已經建好的道路,我的思路就是將建好道路的節點添加到並查集中。

AC代碼:

#include<cstdio>

using namespace std;

typedef struct
{
    int u;
    int v;
    int w;
} edge;

void quicksort(edge a[], int left, int right)
{
    int i, j, temp;
    edge t;

    i = left;
    j = right;
    temp = a[i].w;

    if(i > j)
    {
        return;
    }

    while(i < j)
    {
        while(a[j].w >= temp && i < j)
            j--;
        while(a[i].w <= temp && i < j)
            i++;
        if(i < j)
        {
            t = a[i];
            a[i] = a[j];
            a[j] = t;
        }
    }

    t = a[i];
    a[i] = a[left];
    a[left] = t;

    quicksort(a, left, i-1);
    quicksort(a, j+1, right);
}

void init(int a[], int n)
{
    int i;
    for(i = 1; i <= n; i++)
    {
        a[i] = i;
    }
}

int getf(int a[], int v)
{
    if(a[v] == v)
    {
        return v;
    }
    else
    {
        a[v] = getf(a, a[v]);
        return a[v];
    }
}

int merge(int a[], int u, int v)
{
    int t1 = getf(a, u);
    int t2 = getf(a, v);
    if(t1 == t2)
    {
        return 0;
    }
    else
    {
        a[t2] = t1;
        return 1;
    }
}

int main()
{
    int n, m;
    while(scanf("%d", &n) != EOF && n != 0)
    {
        if(n == 0)
        {
            break;
        }
        int a[105], count = 0, sum = 0;
        int i, tt;
        edge e[6005];
        init(a, 104);
        n = n*(n-1)/2;
        for(i = 1; i <= n; i++)
        {
            scanf("%d%d%d", &e[i].u, &e[i].v, &e[i].w);
            scanf("%d", &tt);
            if(tt == 1)
            {
                merge(a, e[i].u, e[i].v);
            }
        }
        quicksort(e, 1, n);
        for(i = 1; i <= n; i++)
        {
            if(merge(a, e[i].u, e[i].v))
            {
                count++;
                sum += e[i].w;
            }
            if(count == m-1)
            {
                break;
            }
        }

        printf("%d\n", sum);
    }

    return 0;
}

 

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