最小生成树1：Prim算法

Prime算法

  普里姆算法(Prim)，图论中的一种算法，可在加权连通图里搜索最小生成树。意即由此算法搜索到的边子集所构成的树中，不但包括了连通图里的所有顶点，且其所有边的权值之和亦为最小。Prim算法的时间复杂度为O(V²）。
  算法简单描述：
  1).输入：一个加权连通图。其中，顶点集合为V，边集合为E；
  2).初始化：V_new = {x} 其中，x为集合V中的任一节点(起始点)。E_new = {} 为空；
  3).重复下列操作，直到 V_new = V：
    a.在集合E中选取权值最小的边<u, v>。其中，u为集合V_new中的元素，而v不在V_new集合当中，并且v∈V（如果存在有多条满足前述条件即具有相同权值的边，则可任意选取其中之一）；
    b.将v加入集合V_new中，将<u, v>边加入集合E_new中；
  4).输出：使用集合V_new和E_new来描述所得到的最小生成树。

简单证明prim算法。反证法：假设prim生成的不是最小生成树
  1).设prim生成的树为G₀。
  2).假设存在G_min使得cost(G_min)<cost(G₀) 则在G_min中存在<u,v>不属于G₀。
  3).将<u,v>加入G₀中可得一个环，且<u,v>不是该环的最长边(这是因为<u,v>∈G_min)。
  4).这与prim每次生成最短边矛盾。
  5).故假设不成立，命题得证。

图例	说明	不可选	可选	已选 (Vnew)
	此为原始的加权连通图。每条边一侧的数字代表其权值。	-	-	-
	顶点D被任意选为起始点。顶点A、B、E和F通过单条边与D相连。A是距离D最近的顶点，因此将A及对应边AD以高亮表示。	C, G	A, B, E, F	D
	下一个顶点为距离D或A最近的顶点。B距D为9，距A为7，E为15，F为6。因此，F距D或A最近，因此将顶点F与相应边DF以高亮表示。	C, G	B, E, F	A,D
	算法继续重复上面的步骤。距离A为7的顶点B被高亮表示。	C	B,E,G	A,D,F
	在当前情况下，可以在C、E与G间进行选择。C距B为8，E距B为7，G距F为11。E最近，因此将顶点E与相应边BE高亮表示。	无	C,E,G	A,D,F,B
	可供选择的顶点只有C和G。C距E为5，G距E为9，故选取C，并与边EC一同高亮表示。	无	C,G	A, D, F, B, E
	顶点G是唯一剩下的顶点，它距F为11，距E为9，E最近，故高亮表示G及相应边EG。	无	G	A, D, F, B, E, C
	现在，所有顶点均已被选取，图中绿色部分即为连通图的最小生成树。在此例中，最小生成树的权值之和为39。	无	无	A, D, F, B, E, C, G

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int maxn=1005;
const int INF=0x3f3f3f3f;
int p[maxn]; //储存父节点
int edge[maxn][maxn];//邻接矩阵
int d[maxn]; //到生成树的最短距离
int vis[maxn];//是否加入集合中
int n,m;
//初始化
void init()
{
    memset (vis,0,sizeof(vis));
    memset (p,-1,sizeof(p));
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
             i==j? edge[i][j]==0:edge[i][j]=INF;
}
void prim()
{
    //以1作为根节点
    for (int i=1;i<=n;i++)
    {
        if(edge[1][i]!=INF)
            p[i]=1;
        d[i]=edge[1][i];
    }
    while (1)
    {
        int maxx=INF;
        int u=-1;
        //找到距离生成树最短距离的节点
        for (int i=1;i<=n;i++)
        {
            if(!vis[i]&&d[i]<maxx)
            {
                maxx=d[i];
                u=i;
            }
        }
        if(u==-1)
            break;
        //将选出的节点纳入集合
        vis[u]=1;
        //更新该节点影响的节点
        for (int i=1;i<=n;i++)
        {
            if(!vis[i]&&d[i]>edge[u][i])
            {
                d[i]=edge[u][i];
                p[i]=u;
            }
        }
    }
    int sum=0;
    for (int i=1;i<=n;i++)
        if(p[i]==-1)
        {
            printf("不存在最小生成树\n");
            return ;
        }
        else
            sum+=d[i];
        printf("最短生成树的长度为%d\n",sum);
        return ;
}
void parent ()
{
    for (int i=1;i<=n;i++)
        if(p[i]!=-1&&p[i]==i)
              printf("%d为根节点\n",i);
        else if(p[i]!=-1)
              printf("%d的父节点为%d\n",i,p[i]);
}
int main()
{
    scanf("%d%d",&n,&m);
    init();
    for (int i=1;i<=m;i++)
    {
        int x,y,sp;
        scanf("%d%d%d",&x,&y,&sp);
        edge[x][y]=edge[y][x]=sp;
    }
    prim();
    parent();
    return 0;
}

//A(1),B(2),C(3),D(4),E(5),F(6),G(7)
7 11
1 2 7
1 4 5
2 4 9
2 3 8
2 5 7
3 5 5
4 5 15
4 6 6
5 6 8
5 7 9
6 7 11

最短生成树的长度为39
1为根节点
2的父节点为1
3的父节点为5
4的父节点为1
5的父节点为2
6的父节点为4
7的父节点为5