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

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

syn 負責 nod using 通過 als 計算機 esp register

【題目描述】
學校有n臺計算機,為了方便數據傳輸,現要將它們用數據線連接起來。兩臺計算機被連接是指它們有數據線連接。由於計算機所處的位置不同,因此不同的兩臺計算機的連接費用往往是不同的。

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

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

【輸入】
第一行為整數n(2≤n≤100),表示計算機的數目。此後的n行,每行n個整數。第x+1行y列的整數表示直接連接第x臺計算機和第y臺計算機的費用。

【輸出】
一個整數,表示最小的連接費用。

【輸入樣例】
3
0 1 2
1 0 1
2 1 0
【輸出樣例】
2
【提示】
註:表示連接1和2,2和3,費用為2。
# include <bits/stdc++.h>
using namespace std ;
const int N = 100 + 5 ;
int n ;
int fa[N] ;
const int M = N * N ;
struct node {
    int u , v , w ;
}edge[M] ;

inline int find(int x) {
    return x == fa[x] ? x : fa[x] = find(fa[x]) ;
}
inline void merge(int x , int y) {
    fa[find(x)] = find(y) ;
}
inline bool cmp(node x , node y) {
    return x.w < y.w ;
}
signed main() {
    ios::sync_with_stdio(false) ;
    cin >> n ;
    int cnt = 0 ;
    for(register int i=1;i<=n;i++) fa[i] = i ;
    for(register int i=1;i<=n;i++) 
        for(register int j=1;j<=n;j++) {
            int w ;
            cin >> w ;
            edge[++ cnt].u = i , edge[cnt].v = j , edge[cnt].w = w ;
        }
    sort(edge + 1 , edge + cnt + 1 , cmp) ;
    int ans = 0 ;
    for(register int i=1;i<=cnt;i++) {
        int fx = find(edge[i].u) , fy = find(edge[i].v) ;
        if(fx == fy) continue ;
        merge(edge[i].u , edge[i].v) ;
        ans += edge[i].w ;
    }
    printf("%d \n" , ans) ;
    return 0 ;
}

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