【SSL】2021.12.24 1612.最优布线问题

原题网址

由于某些原因，这个网址会进不去…

题目描述

学校有$n$台计算机，为了方便数据传输，现要将它们用数据线连接起来。两台计算机被连接是指它们之间有数据线连接。由于计算机所处的位置不同，因此不同的两台计算机的连接费用往往是不同的。
当然，如果将任意两台计算机都用数据线连接，费用将是相当庞大的。为了节省费用，我们采用数据的间接传输手段，即一台计算机可以间接的通过若干台计算机（作为中转）来实现与另一台计算机的连接。
现在由你负责连接这些计算机，你的任务是使任意两台计算机都连通（不管是直接的或间接的）。

格式

输入格式

第一行为整数$n$，表示计算机的数目。此后的$n$行，每行$n$个整数。第$x+1$行$y$列的整数表示直接连接第$x$台计算机和第$y$台计算机的费用。

输出格式

一个整数，表示最小的连接费用。

样例

输入样例

3
0 1 2
1 0 1
2 1 0

输出样例

2

提示

$2\le n\le 100$
用

memset($array_name$,$value$,sizeof($array_name$)); // 需要 #include<cstring>

给整个数组赋值（赋值$\infty$时$value$为$0x7f$）。

解题思路

最小生成树的板子。

Code

Prim $O(n^2)$

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100;
int n,g[maxn+5][maxn+5],minn[maxn+5],v[maxn+5],ans;
void fileio() {
    freopen("1612.in","r",stdin);
    freopen("1612.out","w",stdout);
}
int main() {
    // fileio();
    scanf("%d",&n);
    for(int i=1;i<=n;i++) {
        for(int j=1;j<=n;j++) {
            scanf("%d",&g[i][j]);
        }
    }
    memset(minn,0x7f,sizeof(minn));
    minn[1]=0;
    memset(v,1,sizeof(v));
    for(int i=1;i<=n;i++) {
        int q=0;
        for(int j=1;j<=n;j++) {
            if(v[j]&&minn[j]<minn[q]) {
                q=j;
            }
        }
        v[q]=0;
        for(int j=1;j<=n;j++) {
            if(v[j]&&g[q][j]<minn[j]) {
                minn[j]=g[q][j];
            }
        }
    }
    for(int i=1;i<=n;i++) {
        ans+=minn[i];
    }
    printf("%d",ans);
	return 0;
}

Kruskal $O(n\times lo{g}_{2}n)$

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn=100;
struct line {
    int x,y,v;
};
vector<line> lines;
int n,f[maxn+5],ans,cnt;
int find(int x) {
    return f[x]==x?x:f[x]=find(f[x]);
}
void unionn(int x,int y) {
    int xt=find(x),yt=find(y);
    if(xt<yt) f[xt]=yt;
    else f[yt]=xt;
}
bool cmp(line x,line y) {
    return x.v<y.v;
}
void fileio() {
    freopen("1612.in","r",stdin);
    freopen("1612.out","w",stdout);
}
int main() {
    // fileio();
    scanf("%d",&n);
    for(int i=1;i<=n;i++) {
        f[i]=i;
        for(int j=1,temp;j<=n;j++) {
            scanf("%d",&temp);
            if(temp) {
                lines.push_back((line){i,j,temp});
            }
        }
    }
    sort(lines.begin(),lines.end(),cmp);
    for(int i=0;i<lines.size();i++) {
        if(find(lines[i].x)!=find(lines[i].y)) {
            cnt++;
            unionn(lines[i].x,lines[i].y);
            ans+=lines[i].v;
            if(cnt==n-1) break;
        }
    }
    printf("%d",ans);
	return 0;
}