p.s.來玩一個XDD
Minimum Spanning Tree(MST)可以通過Prim 和 Kruskal算法來實現
Prim
簡單點來說就是由’點’出發
以這張圖為例吧
我們以華山作為起點
我們可以走到武當山(2)、嵩山(8)和恆山(5)
其中因為走到武當山的路程最短因此我們先把武當山納入我們的領土
很好,現在我們可以走到嵩山(8或7)和恆山(5)
因為去恆山的路程較短所以我們先去征服恆山
這樣我們現在可以去到的地方就有
嵩山(8,7,6)和黑木崖(3)了
既然去黑木崖的路程最短那我們就先去那吧~
所以我們現在可以去到的地方就有
嵩山(8,7,6,6)和泰山(11)了
其中去嵩山的路程較短,由恆山出發和由黑木崖出發的路程都一樣,所以就隨便選吧~
(注:最小生成樹可有多種結果)
距離征服世界的目標不遠了(誤
現在只剩下泰山(11,9)和衡山(6)了
所以我們就先搞定衡山
最後剩下泰山(11,9,6)
其中由衡山出發的路往最短,所以最終會變成這樣
恭喜勇者成功征服世界(誤
實作
現在偶們開個數組來存圖哈
int map[501][501];
然後再開一個數組dis,代表i 點到各點的距離
和數組visit,代表i點是否加入生成樹
int dis[501];
好吧,我們選擇一個點開始加入生成樹
這裡以1號點為例
1.首先要初始化dis數組和visit數組
visit代表生成樹中有哪些點
dis[1] = 0;
for(int i = 2; i <= n; i++){
dis[i] = map[1][i];
}
memset(visit, false, sizeof(visit));
2.然後找出與1號點相連的點中距離最小的點,並將其加入生成樹
visit[1] = true;
for(int i = 1; i <= n; i++){
if(visit[i] == false && dis[i] < min_num){
min_num = dis[i];
j = i; //找出與1號點相連的點中距離最小的點
}
}
visit[j] = true;
3.最後更新與生成樹中的點相連的邊
for(int i = 1; i <= n; i++){
if(visit[i] == false && dis[i] > map[j][i]){
dis[i] = map[j][i];
}
}
4.最後的最後只要重複步驟2和3直到生成樹中有n個點為止
Code
#define MAX_NUM 0x3f3f3f3f
int map[501][501];
int dis[501];
bool visit[501];
int n, cnt, min_num, j, sum;
void prim(){
for(int i = 2; i <= n; i++){
dis[i] = map[1][i];
}
memset(visit, false, sizeof(visit));
sum = 0;
visit[1] = true;
dis[1] = 0;
cnt = 1;
while(cnt < n){
min_num = MAX_NUM;
for(int i = 1; i <= n; i++){
if(visit[i] == false && dis[i] < min_num){
min_num = dis[i];
j = i;
}
}
visit[j] = true;
cnt++;
sum += dis[j];
for(int i = 1; i <= n; i++){
if(visit[i] == false && dis[i] > map[j][i]){
dis[i] = map[j][i];
}
}
}
}
Kruskal
簡單點來說就是由’邊’出發
我們繼續以這幅圖為例
我們把每條邊排序
邊 | 距離 |
---|---|
武當- 華山 | 2 |
恆山-黑木崖 | 3 |
華山-恆山 | 5 |
恆山-嵩山 | 6 |
嵩山-衡山 | 6 |
黑木崖-嵩山 | 6 |
衡山-泰山 | 6 |
武當-嵩山 | 7 |
華山-嵩山 | 8 |
嵩山-泰山 | 9 |
黑木崖-泰山 | 11 |
把每條邊按序加入生成樹
當時加入的邊不能與生成樹中的邊形成「迴圈」
那甚樣知道有木有形成「迴圈」呢?
這時候用并查集(Union Find)就可以啦
所以我們就開個表儲存邊和距離
vector<pair<pair<int, int>, int > >adj; //((頂點1, 頂點2), cost)
因為要用到并查集所以需要再開個數組儲存上級節點
int pre[100];
這是并查集的代碼
int find(int x){
int r = x;
while(pre[r] != r){
r = pre[r];
}
int i = x;
int j;
while(pre[i] != r){ //路徑壓縮
j = pre[i];
pre[i] = r;
i = j;
}
return r;
}
void join(int x, int y){
int p1 = find(x);
int p2 = find(y);
if(p1 != p2){
pre[p1] = p2;
}
}
實現Kruskal
bool cmp(pair<pair<int, int>, int > &a, pair<pair<int, int>, int >&b){
return a.second < b.second;
}
int kruskal(){
int ret = 0;
sort(adj.begin(), adj.end(), cmp); //排序邊
for(int i = 0; i < n; i++) //并查集初始化
pre[i] = i;
for(int i = 0; i < adj.size(); ++i){
int cost = adj[i].second;
int u = adj[i].first.first;
int v = adj[i].first.second;
if(find(u) == find(v)) //形成迴圈就跳過
continue;
join(u, v); //加入并查集
ans.push_back(make_pair(u,v));
ret += cost;
}
return ret; //回傳生成樹的和
}