Description
学校有n台计算机,为了方便数据传输,现要将它们用数据线连接起来。两台计算机被连接是指它们之间有数据线连接。由于计算机所处的位置不同,因此不同的两台计算机的连接费用往往是不同的。
当然,如果将任意两台计算机都用数据线连接,费用将是相当庞大的。为了节省费用,我们采用数据的间接传输手段,即一台计算机可以间接的通过若干台计算机(作为中转)来实现与另一台计算机的连接。
现在由你负责连接这些计算机,你的任务是使任意两台计算机都连通(不管是直接的或间接的)。
Input
输入文件wire.in,第一行为整数n(2<=n<=100),表示计算机的数目。此后的n行,每行n个整数。第x+1行y列的整数表示直接连接第x台计算机和第y台计算机的费用。
Output
输出文件wire.out,一个整数,表示最小的连接费用。
Sample Input
3
0 1 2
1 0 1
2 1 0
Sample Output
2(注:表示连接1和2,2和3,费用为2)
解题思路
很明显的最小生成树模板
prim算法
蓝白点思想,没有加入最小生成树的点看作蓝点,加入了的点看作白点。初始所有点为无限大
于是,因为一开始第一个点为起点dis[1]=0,所以第一点变白点,dis[1]=0 dis[2]=2 dis[3]=4 dis[4]=7 dis[5]=∞
因为dis[2]是所有蓝点中最小的,所以第二点变白点,dis[3]=1 dis[4]=7 dis[5]=2
因为dis[3]是所有蓝点中最小的,所以第三点变白点,dis[4]=1 dis[5]=2
。。。。。。(暴躁省略)
顺便推一下kruskal算法
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int INF=0x7fffffff;
int v[200],n,a[200][200],Gun,dis[200];
void prim(){
memset(dis,0x7f,sizeof(dis));//无穷大
dis[1]=0;//固定起点1
for(int i=1;i<=n;i++){
int ins=INF,k;
for(int j=1;j<=n;j++)
if(dis[j]<ins&&!v[j])//找蓝点最小
ins=dis[j],k=j;
v[k]=1;//变白点
Gun+=dis[k];//累计答案
for(int j=1;j<=n;j++)
if((a[k][j]<dis[j])&&!v[j]&&a[k][j])//更新每个蓝点连接最小生成树的边
dis[j]=a[k][j];
}
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
scanf("%d",&a[i][j]);
prim();
printf("%d",Gun);
}