Prim算法

/*
Prim算法的基本思想
1.清空生成树，任取一个顶点加入生成树
2.在那些一个端点在生成树里，另一个端点
  不在生成树里的边中，选取一条权最小的边，
  将它和另一个端点加进生成树
3.重复步骤2，直到所有的顶点都进入了生成树为止，
  此时的生成树就是最小生成树
注意：prim算法最小生成树的生成过程中，也有可能构成
  回路，需做判断。
*/
#define MAX 1000
#define INF 100000
int graph[MAX][MAX];  //graph[i][j]表示点i到j的距离
int prim(int graph[][MAX],int n)
{
    int i,j,l,min1,k;
    int lowcost[MAX],visit[MAX];
    int sum=0;
    for(int i=2; i<=n; i++)
    {
        lowcost[i]=graph[1][i];//初始化
        visit[i]=1; //点未加入最小生成树中
    }
    lowcost[1]=0;//标记顶点1加入最小生成树中
    for(int i=2; i<=n; i++)
    {
        min1=INF;
        k=0;
        for(int j=2; j<=n; j++)
            if((lowcost[j]<min1)&&(visit[i]!=0))
            {
                min1=lowcost[j];
                k=j;
            }
    }
    visit[k]=0;
    sum+=min1;
    for(int j=0; j<=n; j++)
    {
        if(visit[i]==1&&lowcost[j]>graph[k][j])
            lowcost[j]=graph[k][j];
    }
    return sum;
}

int prime(int cur)     //以cur这个点开始建立最小生成树
{
    int index;
    int sum = 0;
    memset(visit, false, sizeof(visit));
    visit[cur] = true;            //标记已进入最小生成树中
    for(int i = 0; i < m; i ++)
    {
        dist[i] = graph[cur][i];   //初始化每个点到cur这个点的距离
    }
    for(int i = 1; i < m; i ++)
    {
        int mincost = INF;
        for(int j = 0; j < m; j ++)
        {
            if(!visit[j] && dist[j] < mincost)
            {
                mincost = dist[j];
                index = j;
            }
        }
        visit[index] = true;
        sum += mincost;
        for(int j = 0; j < m; j ++)
        {
            if(!visit[j] && dist[j] > graph[index][j])
            {
                dist[j] = graph[index][j];
            }
        }
    }
    return sum;
}