数据结构基础58-最优布线

题目描述

学校有n台计算机，为了方便数据传输，现要将它们用数据线连接起来。两台计算机被连接是指它们间有数据线连接。由于计算机所处的位置不同，因此不同的两台计算机的连接费用往往是不同的。

当然，如果将任意两台计算机都用数据线连接，费用将是相当庞大的。为了节省费用，我们采用数据的间接传输手段，即一台计算机可以间接的通过若干台计算机（作为中转）来实现与另一台计算机的连接。

  现在由你负责连接这些计算机，任务是使任意两台计算机都连通（不管是直接的或间接的）。

输入

输入文件wire.in，第一行为整数n（2<=n<=100），表示计算机的数目。此后的n行，每行n个整数。第x+1行y列的整数表示直接连接第x台计算机和第y台计算机的费用。

输出

输出文件wire.out，一个整数，表示最小的连接费用。

样例输入

3

0 1 2

1 0 1

2 1 0

样例输出

2

提示

（注：表示连接1和2，2和3，费用为2）

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

struct P{
    int u,v,w;
};

bool cmp(P a,P b){
    return a.w<b.w;
}

const int N = 105;
int n,f[N],ans=0;

int getfa(int x){
    if(f[x]==x){
        return x;
    }else{
        return f[x]=getfa(f[x]);
    }
}

void merge(int x,int y){
    int fx=getfa(x),fy=getfa(y);
    if(fx==fy)return;
    f[fx]=fy;
}

vector<P>pd;

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin>>n;
    for(int i=1;i<=n;i++)f[i]=i;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            int w;
            cin>>w;
            P t;
            t.u=i;t.v=j;t.w=w;
            pd.push_back(t);
        }
    }
    sort(pd.begin(),pd.end(),cmp);
    for(int i=0;i<pd.size();i++){
        int p=getfa(pd[i].u),q=getfa(pd[i].v);
        if(p!=q){
            ans+=pd[i].w;
            merge(p,q);
        }
    }
    cout<<ans;
}