Prim算法模板

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int N=1001;
const int inf=1<<29;
int w[N][N];
int dis[N],flag[N];
int n,m,u,v,c;
int  prim()
{
    int sum=0;//计算最小距离
    memset(flag,0,sizeof(flag));
    for(int i=1; i<=n; i++)
    {
        dis[i]=w[1][i];//把起点到每个点的距离付给dis
    }
    flag[1]=1;
    for(int i=1; i<n; i++)//会更新n-1次顶点，每个点都会做一次顶点
    {
        int to=-1,min1=inf;
        for(int j=1; j<=n; j++)//从起点开始找，求出各点到以找过的顶点的距离最小值
        {
            if(!flag[j]&&dis[j]<min1)//保证不是被使用过的顶点，重新算出此次循环的最小值
            {
                to=j;//把最小值对应的点的下标记录下
                min1=dis[j];
            }
        }
        if(to==-1) return -1;//有断点，不能到达
        sum+=min1;
        flag[to]=1;
        for(int j=1; j<=n; j++)
        {
            dis[j]=min(dis[j],w[to][j]);//更新顶点，其中存储着最短的目前找过的顶点与各点之间的距离
        }//这样方便以后可以直接找到与各点的最短距离，遇到回路也不怕！
    }
    return sum;

}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)//n个顶点，m条边
    {
        for(int i=1; i<=n; i++) //初始化i到j点的距离为无穷大，
        {
            for(int j=1; j<=n; j++)
            {
                w[i][j]=inf;//i到j，j到i 都要人为的付成是一样的
            }
        }
        for(int i=0; i<m; i++)
        {
            scanf("%d%d%d",&u,&v,&c);//起点u，终点v，权值c
            w[u][v]=w[v][u]=c;
        }
        printf("%d
",prim());
    }
    return 0;
}