ssl 1612 最优布线问题#prim#

题目

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

分析

求最小生成树。

---

代码

#include <cstdio>
#include <cctype>
#include <climits>
using namespace std;
int a[101][101],lowcost[101]; bool v[101];
inline int in(){
	char x=getchar(); int ans=0;
	while (!isdigit(x)) x=getchar();
	while (isdigit(x)) ans=ans*10+x-48,x=getchar();
	return ans;
}
int main(){
	int n,k,min,s=0; n=in();
	for (int i=1;i<=n;i++){
		for (int j=1;j<=n;j++) a[i][j]=in();
		lowcost[i]=a[1][i];
	} v[1]=1;
	for (int i=1;i<=n-1;i++){
		min=INT_MAX;
		for (int j=1;j<=n;j++)
		if (lowcost[j]<min&&!v[j]){
			min=lowcost[j];
			k=j;
		}
		s+=min;
		v[k]=1;
		for (int j=1;j<=n;j++) if (a[k][j]<lowcost[j]) lowcost[j]=a[k][j];
	}
	printf("%d",s); return 0;
}