阳江市CSP-J/S 信奥陈老师解题：1350：【例4-11】最短网络(agrinet)

【题目描述】

农民约翰被选为他们镇的镇长！他其中一个竞选承诺就是在镇上建立起互联网，并连接到所有的农场。当然，他需要你的帮助。约翰已经给他的农场安排了一条高速的网络线路，他想把这条线路共享给其他农场。为了用最小的消费，他想铺设最短的光纤去连接所有的农场。你将得到一份各农场之间连接费用的列表，你必须找出能连接所有农场并所用光纤最短的方案。每两个农场间的距离不会超过100000100000。

【输入】

第一行：农场的个数，N（3≤N≤100）N（3≤N≤100）。

第二行..结尾:后来的行包含了一个N×NN×N的矩阵,表示每个农场之间的距离。理论上，他们是NN行，每行由NN个用空格分隔的数组成，实际上，他们限制在8080个字符，因此，某些行会紧接着另一些行。当然，对角线将会是00，因为不会有线路从第ii个农场到它本身。

【输出】

只有一个输出，其中包含连接到每个农场的光纤的最小长度。

【输入样例】

4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0

【输出样例】

28

#include <bits/stdc++.h>
using namespace std;
int n,g[101][101],ans=0,INF=0x3f3f3f3f;
struct Node
{
	int adj;
	int cost;
}c[101];
int getMin()
{
	int pos=0,minn=INF;
	for(int i=1;i<=n;i++)
	{
		if(c[i].cost!=0&&c[i].cost<minn)
		{
			minn=c[i].cost;
			pos=i;
		}
	}
	return pos;
}
void Prim(int u)
{
	for(int j=1;j<=n;j++)
	{
		if(j!=u)
		{
			c[j].adj=u;
			c[j].cost=g[u][j]; //存放连着i的顶点以及最小路径
		}
	}
	c[u].cost=0;
	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()
{
	cin>>n;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		{
			cin>>g[i][j];
			if(g[i][j]==0)
				g[i][j]=INF;
		}
	Prim(1);
	cout<<ans;
	return 0;
}