1. 程式人生 > 其它 >1349:【例4-10】最優佈線問題

1349:【例4-10】最優佈線問題

【題目描述】

學校有nn臺計算機,為了方便資料傳輸,現要將它們用資料線連線起來。兩臺計算機被連線是指它們有資料線連線。由於計算機所處的位置不同,因此不同的兩臺計算機的連線費用往往是不同的。

當然,如果將任意兩臺計算機都用資料線連線,費用將是相當龐大的。為了節省費用,我們採用資料的間接傳輸手段,即一臺計算機可以間接的通過若干臺計算機(作為中轉)來實現與另一臺計算機的連線。

現在由你負責連線這些計算機,任務是使任意兩臺計算機都連通(不管是直接的或間接的)。

【輸入】

第一行為整數nn(2n1002≤n≤100),表示計算機的數目。此後的nn行,每行nn個整數。第x+1x+1行yy列的整數表示直接連線第

xx臺計算機和第yy臺計算機的費用。

【輸出】

一個整數,表示最小的連線費用。

【輸入樣例】

3
0 1 2
1 0 1
2 1 0

【輸出樣例】

2

【提示】

注:表示連線11和22,22和33,費用為2

#include <bits/stdc++.h>
using namespace std;

struct Node { // 邊[x,y]權重w
    int x, y, w;

    Node(int x, int y, int w)
    {
        this->x = x;
        this->y = y;
        this->w = w;
    }
};

bool cmp(const Node &a, const Node &b)
{ // 邊的比較,小的在前
    return (a.w < b.w);
}

void init(vector<int> &da)
{ // 並查集的初始化
    for (int i = 0; i < da.size(); i++) {
        da[i] = i;
    }
}

int find(vector<int> &da, int x)
{ // 並查集的查詢
    if (da[x] != x) {
        da[x] = find(da, da[x]);
    }
    return da[x];
}

bool united(vector<int> &da, int x, int y)
{ // 並查集的判斷
    int xda = find(da, x);
    int yda = find(da, y);
    return (xda == yda);
}

void unite(vector<int> &da, int x, int y)
{ // 並查集的合併
    int xda = find(da, x);
    int yda = find(da, y);
    if (xda != yda) {
        da[xda] = yda;
    }
}

int kruskal(vector<Node> &es, int n)
{ // 克魯斯卡爾演算法
    int ans = 0; // 答案
    vector<int> da(n + 1); // 並查集
    init(da); // 初始化並查集
    sort(es.begin(), es.end(), cmp);
    for (auto &e : es) { // 遍歷所有邊
        if (united(da, e.x, e.y)) {
            continue; // 能形成環
        }
        // cout << e.x << "," << e.y << endl;
        unite(da, e.x, e.y);
        ans += e.w;
    }
    return ans;
}

int main()
{
    // freopen("in.txt", "r", stdin);
    int n, w; // 節點數n, 權重w
    scanf("%d", &n);
    vector<Node> es; // 邊集
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) {
            scanf("%d%", &w);
            if (w > 0 && i < j) { // 非0
                es.push_back(Node(i, j, w));
            }
        }
    }
    int ans = kruskal(es, n);
    printf("%d\n", ans);
    return 0;
}