1349:【例4-10】最優佈線問題
阿新 • • 發佈:2021-11-21
【題目描述】
學校有nn臺計算機,為了方便資料傳輸,現要將它們用資料線連線起來。兩臺計算機被連線是指它們有資料線連線。由於計算機所處的位置不同,因此不同的兩臺計算機的連線費用往往是不同的。
當然,如果將任意兩臺計算機都用資料線連線,費用將是相當龐大的。為了節省費用,我們採用資料的間接傳輸手段,即一臺計算機可以間接的通過若干臺計算機(作為中轉)來實現與另一臺計算機的連線。
現在由你負責連線這些計算機,任務是使任意兩臺計算機都連通(不管是直接的或間接的)。
【輸入】
第一行為整數nn(2≤n≤1002≤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; }