Constructing Roads - HDU 1102 最小生成树

题意：给你N*N矩阵，表示N个点之间的距离。FJ要把N个点全都连接起来，求连接的最短距离。（即传说中的最小生成树），另外此题还有一个条件就是现在某些点已经连接。

 个人思路：

        该题为Agri-Net - POJ 1258 题的变形，其实差不多，只不过POJ1258题没有后一个条件。该题每给出一个m1，m2两点，让矩阵a[m1][m2]=0,a[m2][m1]=0就可以解决了，详情看一下我的Agri-Net - POJ 1258就可以了   http://blog.csdn.net/u014733623/article/details/23866901

        或者我在这先copy一下吧，先假设没有后一个条件：

        a[][]为输入的矩阵,设一个数组d[]表示当前已连接的点的集合到还未连接的点的最小距离，每次的循环找到离当前已连接的点距离最近的点，对于最开始的时候让d[1]=0，其他为无穷大，使第一次循环的时候连接第一个点，（这句话看不懂可以先继续往下看），然后你现在已经连接了第一个点，那么你再连到其他点的距离就应该是a[1][j]，将其输入d[]中。第二次循环，找到d[]中最短的点，现在你已连接的点又多了一个，那么你这两个点再连接其他点的最短距离可能会更短，用d[j]=min(d[j],a[pos][j])修改d[]中的值，一直循环n次，每次连接1个点，直到连接了所有的点结束。

        然后该题每给出一个m1，m2两点，让矩阵a[m1][m2]=0,a[m2][m1]=0，让其先预处理一下再执行上述的循环，因为已连接的点距离为0，所以会先连接该点，对结果也没有影响。

        PS：这题有点坑，就是上面没说有多组数据，但是你在HDU上提交的话，它会给你测多组数据，所以需要注意一下。

下面是我的AC代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

int a[1010][1010];     //a[][]为输入的矩阵
int n,ans;
int d[1010];           //d[]为当前已连接的点的集合到还未连接的点的最小距离
int visit[1010];       //visit[]标记已经连接的点
const int INF=100000000;
void Prime()
{ int i,j;
  for(i=1;i<=n;i++)
   d[i]=INF;
  d[1]=0;              //对于最开始的时候让d[1]=0，其他为无穷大，使第一次循环的时候连接第一个点
  memset(visit,0,sizeof(visit));
  for(i=1;i<=n;i++)
  { int pos=INF,mi=INF; //pos为d[]表示的距离中距离最短的点的指针，mi表示最短的距离
    for(j=1;j<=n;j++)   //找到该点
     if(!visit[j] && d[j]<=mi) //visit[]记录的已连接的点就不用考虑了
     { pos=j;
       mi=d[j];
     }
    visit[pos]=1;       //该点已连接
    ans+=mi;            //总的长度加上这个最短距离
    for(j=1;j<=n;j++)   //现在你已连接的点（第pos个点）又多了一个，那么你再连接其他点的最短距离可能会更短
     if(!visit[j])
     d[j]=min(d[j],a[pos][j]);
  }
}
int main()
{ int i,j,k,num,r1,r2;
  while(~scanf("%d",&n))
  { for(i=1;i<=n;i++)
     for(j=1;j<=n;j++)
      scanf("%d",&a[i][j]);
    ans=0;
    scanf("%d",&num);
    for(i=1;i<=num;i++)    //对于已连接的两个点让其互相的距离为0
    { scanf("%d",&r1);
      scanf("%d",&r2);
      a[r1][r2]=0;
      a[r2][r1]=0;
    }
    Prime();
    printf("%d\n",ans);
  }
}