Prim算法
Prim
算法是一种产生最小生成树的算法。该算法于1930
年由捷克数学家沃伊捷赫·亚尔尼克(英语:Vojtěch Jarník
)发现;并在1957
年由美国计算机科学家罗伯特·普里姆(英语:Robert C. Prim
)独立发现;1959
年,艾兹格·迪科斯彻再次发现了该算法。
Prim
算法从任意一个顶点开始,每次选择一个与当前顶点集最近的一个顶点,并将两顶点之间的边加入到树中。Prim
算法在找当前最近顶点时使用到了贪婪算法。
算法描述:
1. 在一个加权连通图中,顶点集合V
,边集合为E
2. 任意选出一个点作为初始顶点,标记为visit
,计算所有与之相连接的点的距离,选择距离最短的,标记visit
.
3. 重复以下操作,直到所有点都被标记为visit
:
在剩下的点钟,计算与已标记visit
点距离最小的点,标记visit
,证明加入了最小生成树。
下面我们来看一个最小生成树生成的过程:
1 起初,从顶点a
开始生成最小生成树
2 选择顶点a
后,顶点啊置成visit
(涂黑),计算周围与它连接的点的距离:
3 与之相连的点距离分别为7
,6
,4
,选择C
点距离最短,涂黑C
,同时将这条边高亮加入最小生成树:
4 计算与a,c
相连的点的距离(已经涂黑的点不计算),因为与a
相连的已经计算过了,只需要计算与c
相连的点,如果一个点与a,c
都相连,那么它与a
的距离之前已经计算过了,如果它与c的距离更近,则更新距离值,这里计算的是未涂黑的点距离涂黑的点的最近距离,很明显,b
和a
为7
,b
和c
的距离为6
,更新b
和已访问的点集距离为6
,而f
,e
和c
的距离分别是8
,9
,所以还是涂黑b
,高亮边bc
:
5 接下来很明显,d
距离b
最短,将d
涂黑,bd
高亮:
6 f
距离d
为7
,距离b
为4
,更新它的最短距离值是4
,所以涂黑f
,高亮bf
:
7 最后只有e
了:
针对如上的图,代码实例如下:
- #include<iostream>
- #define INF 10000
- using namespace std;
- constint N = 6;
- bool visit[N];
- intdist[N] = { 0, };
- intgraph[N][N] = { {INF,7,4,INF,INF,INF}, //INF代表两点之间不可达
- {7,INF,6,2,INF,4},
- {4,6,INF,INF,9,8},
- {INF,2,INF,INF,INF,7},
- {INF,INF,9,INF,INF,1},
- {INF,4,8,7,1,INF}
- };
- intprim(intcur)
- {
- intindex = cur;
- intsum = 0;
- inti = 0;
- intj = 0;
- cout << index << " ";
- memset(visit,false, sizeof(visit));
- visit[cur] = true;
- for(i = 0; i < N; i++)
- dist[i] = graph[cur][i];//初始化,每个与a邻接的点的距离存入dist
- for(i = 1; i < N; i++)
- {
- intminor = INF;
- for(j = 0; j < N; j++)
- {
- if(!visit[j] && dist[j] < minor) //找到未访问的点中,距离当前最小生成树距离最小的点
- {
- minor = dist[j];
- index = j;
- }
- }
- visit[index] = true;
- cout << index << " ";
- sum += minor;
- for(j = 0; j < N; j++)
- {
- if(!visit[j] && dist[j]>graph[index][j]) //执行更新,如果点距离当前点的距离更近,就更新dist
- {
- dist[j] = graph[index][j];
- }
- }
- }
- cout << endl;
- returnsum; //返回最小生成树的总路径值
- }
- intmain()
- {
- cout << prim(0) << endl;//从顶点a开始
- return0;
- }