Agri-Net POJ - 1258（最小生成树）详解注释

题目链接->
有n个农场，已知这n个农场都互相相通，有一定的距离，现在每个农场需要装光纤，问怎么安装光纤能将所有农场都连通起来，并且要使光纤距离最小，输出安装光纤的总距离
任意两个村庄之间的距离小于 100,000.
Input
输入包含多组数据。对于每组数据, 第一行包含一个整数N表示农场的数量 (3 <= N <= 100). 接下来是一个NXN的邻接矩阵,表示各个村庄之间的距离 . 当然，对角线将是0，因为从农场到农场的距离对于这个问题并不有趣。
Output
对于每种情况，请输出一个整数表示长度，该长度是连接整个农场集合所需的最小光纤长度的总和。
Sample Input
4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0
Sample Output
28

题意思路：就是把点与点之间的距离用矩阵表示了，套用一下模板就可以了
代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
#define inf 0x3f3f3f3f
//const int inf=0x3f3f3f3f;
int n;
int mp[105][105],dis[105];
//int vis[105];
bool vis[105];//0为false，1为true。
void Prim()
{
    int v,mi;
	/*for(int i=2;i<=n;i++)
    {
		dis[i]=mp[1][i];
		//vis[i]=1;
		vis[i]=false;
	}*/
	for(int i=1;i<=n;i++)
    {
        dis[i]=mp[1][i];//先把1加入Vnew,若给出1到与1相连的点，则dis就是给出的输入的mp[1][i]，否则就是inf
        vis[i]=false;
    }
	//vis[1]=0;
	//vis[1]=true;
	//dis[1]=0;
	int sum=0;
	//for(int i=1;i<=n-1;i++)
    for(int i=1;i<=n;i++)
	{
	    mi=inf;
		for(int j=1;j<=n;j++)
		    //if(mi>dis[j]&&vis[j]!=0)
            if(!vis[j]&&mi>dis[j])//在集合V中先找出一个没有加入集合Vnew且权值最小的点
            {
                mi=dis[j];
                v=j;//v先后等于1,2,3
            }
		vis[v]=true;//标记该点已经加入Vnew
		dis[v]=0;//路已经修好，距离更新为0
		sum+=mi;
		for(int j=1;j<=n;j++)//更新 集合V中的点 到 集合Vnew中的点v 的距离
		{
			/*if(dis[j]>dis[v]+mp[v][j])
			{
				dis[j]=dis[v]+mp[v][j];
				vis[j]=v;
			}*/
			if(!vis[j]&&dis[j]>mp[v][j])//如果这个点没有加入Vnew还在V中 且 加入v点后 j点到v点的距离更近了，
            {
                dis[j]=mp[v][j];//则更新dis中的距离值
            }
		}
	}
	printf("%d\n",sum);
}
int main()
{
	while(scanf("%d",&n)!=EOF)
    {
        for(int i=1;i<=n;i++)
	    for(int j=1;j<=n;j++)
	       //mp[i][j]=mp[j][i]=(i==j)?0:inf;
	       {
	           if(i==j)
                mp[i][j]=mp[j][i]=0;
               else
                mp[i][j]=mp[j][i]=inf;
	       }
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                scanf("%d",&mp[i][j]);
        Prim();
    }
	return 0;
}