最小生成树(prim算法)

        一个有 n 个结点的连通图的生成树是原图的极小连通子图，且包含原图中的所有 n 个结点，并且有保持图联通的最少的边。

假设V是图中顶点的集合,E是图中边的集合,TE为最小生成树中的边的集合,则prim算法通过以下步骤可以得到最小生成树:

 

　　1:初始化:U={u 0},TE={f}。此步骤设立一个只有结点u 0的结点集U和一个空的边集TE作为最小生成树的初始形态,在随后的算法执行中,这个形态会不断的发生变化,直到得到最小生成树为止。

　　

      2:在所有u∈U,v∈V－U的边(u,v)∈E中,找一条权最小的边(u 0,v 0)，将此边加进集合TE中，并将此边的非U中顶点加入U中。此步骤的功能是在边集E中找一条边,要求这条边满足以下条件:首先边的两个顶点要分别在顶点集合U和V－U中,其次边的权要最小。找到这条边以后,把这条边放到边集TE中,并把这条边上不在U中的那个顶点加入到U中。这一步骤在算法中应执行多次,每执行一次,集合TE和U都将发生变化,分别增加一条边和一个顶点,因此,TE和U是两个动态的集合,这一点在理解算法时要密切注意。

　　

      3:如果U=V,则算法结束;否则重复步骤2。可以把本步骤看成循环终止条件。我们可以算出当U=V时,步骤2共执行了n－1次(设n为图中顶点的数目),TE中也增加了n－1条边,这n－1条边就是需要求出的最小生成树的边。

 

#include<iostream> #include<climits> #define MAXSIZE 100 using namespace std; int n; int graph[MAXSIZE][MAXSIZE]; int lowest[MAXSIZE]; int pre[MAXSIZE]; bool state[MAXSIZE]; int main(void) { cout<<"请输入n:"; cin>>n; cout<<"请输入二维数组:"<<endl; for(int i=1;i<=n;i++) { for(int j=1;j<=n;j++) { cin>>graph[i][j]; } } for(int i=2;i<=n;i++) { lowest[i]=graph[1][i]; pre[i]=1; state[i]=false; } state[1]=true; for(int i=1;i<n;i++) { int temp=INT_MAX; int v=1; for(int i=2;i<=n;i++) { if(!state[i]&&lowest[i]<temp&&lowest[i]!=-1) { temp=lowest[i]; v=i; } } state[v]=true; cout<<v<<endl; for(int i=1;i<=n;i++) { if( !state[i] && ( (graph[v][i]<lowest[i]&&graph[v][i]!=-1) || (lowest[i]==-1) ) ) { lowest[i]=graph[v][i]; pre[i]=v; } } } for(int i=2;i<=n;i++) { cout<<i<<' '<<pre[i]<<' '<<endl; } return 0; } /* 6 -1 6 1 5 -1 -1 6 -1 5 -1 3 -1 1 5 -1 5 6 4 5 -1 5 -1 -1 2 -1 3 6 -1 -1 6 -1 -1 4 2 6 -1 */