南沙CSP-J/S家教老师： 1349：【例4-10】最优布线问题

【题目描述】

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

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

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

【输入】

第一行为整数n（2≤n≤1002），表示计算机的数目。此后的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 INF=0x3f3f3f3f;
int g[101][101];
int n,ans=0;
struct Close {
	int adj;  //邻接点 
	int cost; //代价 
};
Close c[101];
int getMin()
{
	int min=INF,pos=0;
	for(int i=1;i<=n;i++)
	{
		if(c[i].cost!=0&&c[i].cost<min)
		{
			min=c[i].cost;
			pos=i;
		}
	}
	return pos;
}

void Prim(int u)
{
	for(int i=1;i<=n;i++)
	{
		if(i!=u)
		{
			c[i].adj=u;
			c[i].cost=g[u][i]; //存放连着U的顶点以及最小路径
		}
	}
	c[u].cost=0;  //初始U={u} 
	for(int i=1;i<=n;i++)
	{
		int pos=getMin();
		if(c[pos].adj!=0)
		{
			ans+=g[pos][c[pos].adj];
		}
		c[pos].cost=0; 
		for(int j=1;j<=n;j++)
		{
			if(c[j].cost!=0&&g[pos][j]<c[j].cost)
			{
				c[j].cost=g[pos][j];
				c[j].adj=pos;
			}
		}
	}
	
}
int main()
{
	memset(g,0x3f,sizeof(g));
	cin>>n;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			cin>>g[i][j];
	Prim(1);
	cout<<ans;
	return 0;
}


​