prim算法

prim算法：

第一步：先任意选择一个点作为  最小生成树 的起始点。

第二步：然后找出所有点中 到 最小生成树 的最小距离的点。又加入最小生成树

第三步：然后再更新 所有点 到最小生成树的 距离。

再重复第二三步。一直到n个点都被加入进来了。

例题:hdu 1102

#include<iostream> #include<cstdio> using namespace std; #define INF 0x1f1f1f1f//定义一个整数常量，表示无穷大 //prim函数返回得到的最小生成树的n-1条边的权值和 //参数cost 为表示图的矩阵，n为顶点个数 int prim(int cost[][200],int n) {     //low表示每个点到生成树的最小距离，vis表示一个点是否已加入生成树中     int low[10000],vis[10000]={0};     int i,j,p;     int min,res=0;     vis[0]=1;     for(i=1;i<n;i++)low[i]=cost[0][i];     for(i=1;i<n;i++)     {         min=INF;p=-1;         for(j=0;j<n;j++)         {             if(0==vis[j]&&min>low[j])             {                 min=low[j];                 p=j;             }         }         //min==INF说明找不到能够加入的点了，说明图是不连通的         if(min==INF)return -1;          res+=min;         vis[p]=1; for(j=0;j<n;j++)         {             if(0==vis[j]&&low[j]>cost[p][j])             {                 low[j]=cost[p][j];             }         }     }     return res; }  int main() {     int n;     int a[200][200]; while(cin>>n)     {         for(int i=0;i<n;i++)         {             for(int j=0;j<n;j++)             {                 scanf("%d",&a[i][j]);             }         }         int q;         scanf("%d",&q);         for(int i=0;i<q;i++)         {             int u,v;             scanf("%d%d",&u,&v);             a[u-1][v-1]=a[v-1][u-1]=0;         }         int res=prim(a,n);         printf("%d\n",res);     }     return 0; }