最小生成树(Prim算法和Kruskal算法)
一、Prim算法
算法思路: 首先就是从图中的一个起点a开始,把a加入U集合,然后,寻找从与a有关联的边中,权重最小的那条边并且该边的终点b在顶点集合:(V-U)中,我们也把b加入到集合U中,并且输出边(a,b)的信息,这样我们的集合U就有:{a,b},然后,我们寻找与a关联和b关联的边中,权重最小的那条边并且该边的终点在集合:(V-U)中,我们把c加入到集合U中,并且输出对应的那条边的信息,这样我们的集合U就有:{a,b,c}这三个元素了,一次类推,直到所有顶点都加入到了集合U。
下面我们对下面这幅图求其最小生成树:
- 假设我们从顶点v1开始,所以我们可以发现(v1,v3)边的权重最小,所以第一个输出的边就是:v1—v3=1:
- 然后,我们要从v1和v3作为起点的边中寻找权重最小的边,首先了(v1,v3)已经访问过了,所以我们从其他边中寻找,发现(v3,v6)这条边最小,所以输出边就是:v3—-v6=4
- 然后,我们要从v1、v3、v6这三个点相关联的边中寻找一条权重最小的边,我们可以发现边(v6,v4)权重最小,所以输出边就是:v6—-v4=2.
- 然后,我们就从v1、v3、v6、v4这四个顶点相关联的边中寻找权重最小的边,发现边(v3,v2)的权重最小,所以输出边:v3—–v2=5
- 然后,我们就从v1、v3、v6、v4,v2这2五个顶点相关联的边中寻找权重最小的边,发现边(v2,v5)的权重最小,所以输出边:v2—–v5=3
- 最后,我们发现六个点都已经加入到集合U了,我们的最小生成树建立完成。
二、Kruskal算法
算法思路: (1)将图中的所有边都去掉。 (2)将边按权值从小到大的顺序添加到图中,保证添加的过程中不会形成环 (3)重复上一步直到连接所有顶点,此时就生成了最小生成树。这是一种贪心策略。
这里同样我们给出一个和Prim算法讲解中同样的例子,模拟克鲁斯卡算法生成最小生成树的详细的过程:
例题——1
HUD-1863 Problem Description 省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。经过调查评估,得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序,计算出全省畅通需要的最低成本。 Input 测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 );随后的 N 行对应村庄间道路的成本,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间道路的成本(也是正整数)。为简单起见,村庄从1到M编号。当N为0时,全部输入结束,相应的结果不要输出。 Output 对每个测试用例,在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通,则输出“?”。 Sample Input 3 3 1 2 1 1 3 2 2 3 4 1 3 2 3 2 0 100 Sample Output 3 ?
1—Prim算法
#include <queue>
#include <iostream>
#include <cstring>
#include <cmath>
#include <ctime>
#include <climits>
#include <vector>
#include <algorithm>
#include <map>
#include <cstdio>
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
#define INF 0x3f3f3f3f
int N,M;
int x,y,z;
int Map[105][105],dist[105];
bool vis[105];
int sum;
void Prim(){
for(int i=1;i<=M;++i){
dist[i]=Map[1][i];
}
memset(vis,0,sizeof(vis));
vis[1]=1;
sum=0;
for(int i=1;i<=M;++i){
int minn=INF;
int cur=-1;
for(int j=2;j<=M;++j){
if(!vis[j]&&dist[j]<minn){
minn=dist[j];
cur=j;
}
}
if(cur==-1){
break;
}
sum+=dist[cur];
vis[cur]=1;
for(int k=1;k<=M;++k){
if(!vis[k]&&dist[k]>Map[cur][k]){
dist[k]=Map[cur][k];
}
}
}
bool flag=true;
for(int i=1;i<=M;++i) if(!vis[i]){ flag=false; break;}
if(flag) printf("%d\n",sum);
else puts("?");
}
int main(){
while(~scanf("%d%d",&N,&M)&&N){
fill(Map[0],Map[0]+105*105,INF);
for(int i=1;i<=N;++i){
scanf("%d%d%d",&x,&y,&z);
if(Map[x][y]>z) Map[x][y]=Map[y][x]=z;
}
Prim();
}
return 0;
}
2——Kruskal算法
#include <queue>
#include <iostream>
#include <cstring>
#include <cmath>
#include <ctime>
#include <climits>
#include <vector>
#include <algorithm>
#include <map>
#include <cstdio>
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
const ULL maxn=2E5+10;
struct node
{
int x,y,z;
}edge[maxn];
bool cmp(node a,node b) {
return a.z < b.z;
}
int n,m;
ULL sum;
int vis[maxn];
void Init(){
for(int i=0;i<=m;++i){ vis[i]=i;}
}
int Find(int n){
return n==vis[n] ? n : vis[n]=Find(vis[n]);
}
void Combine(int a,int b){
int x=Find(a);
int y=Find(b);
if(x==y) return;
vis[y]=x;
}
void Kruskal(){
Init();
sort(edge+1,edge+n+1,cmp);
node point;
sum=0;
for(int i=1;i<=n;++i){
point=edge[i];
if(Find(point.x)!=Find(point.y)){
Combine(point.x,point.y);
sum+=edge[i].z;
}
}
}
int main(){
while(~scanf("%d%d",&n,&m)&&n&&m){
for(int i=1;i<=n;++i){ scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].z);}
Kruskal();
bool flag=true;
for(int i=2;i<=m;++i) {if(Find(1)!=Find(i)) {flag=false; break;}}
if(flag) printf("%lld\n",sum);
else puts("?");
}
return 0;
}
例题——2
HDU-1233 Problem Description 某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
Input 测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。 当N为0时,输入结束,该用例不被处理。
Output 对每个测试用例,在1行里输出最小的公路总长度。
Sample Input 3 1 2 1 1 3 2 2 3 4 4 1 2 1 1 3 4 1 4 1 2 3 3 2 4 2 3 4 5 0
Sample Output 3 5
1——Prim算法
#include <queue>
#include <iostream>
#include <cstring>
#include <cmath>
#include <ctime>
#include <climits>
#include <vector>
#include <algorithm>
#include <map>
#include <bits/stdc++.h>
typedef long long LL;
typedef unsigned long long ULL;
using namespace std;
const int MAXN = 100;
const int INF = 0x3f3f3f3f;
bool vis[MAXN];
int dis[MAXN];
int G[MAXN][MAXN];
int n, m;
void init () {
memset(vis, 0, sizeof(vis));
fill(dis, dis + MAXN, INF);
for(int i = 1; i <= n; ++i) {
for(int j = 1; j <= n; ++j) {
if(i == j) G[i][j] = 0;
else G[i][j] = INF;
}
}
}
void Prime () {
dis[1] = 0;
while (true) {
int k = -1, minn = INF;
for(int i = 1; i <= n; ++i) {
if( dis[i] < minn && !vis[i] ) {
k = i;
minn = dis[i];
}
}
if(k == -1) break;
vis[k] = 1;
for(int i = 1; i <= n; ++i) {
if (dis[i] > G[k][i] && !vis[i]) {
dis[i] = G[k][i];
}
}
}
int ans = 0;
for(int i = 1; i <= n; ++i) ans += dis[i];
cout << ans << endl;
}
int main(){
while (cin >> n && n) {
init();
m = n * (n - 1) / 2;
int x, y, z;
for(int i = 0; i < m; ++i) {
cin >> x >> y >> z;
if( G[x][y] > z) {
G[x][y] = z, G[y][x] = z;
}
}
Prime();
}
return 0;
}
2——Kruskal算法
#include <queue>
#include <iostream>
#include <cstring>
#include <cmath>
#include <ctime>
#include <climits>
#include <vector>
#include <algorithm>
#include <map>
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
const ULL maxn=2E5+10;
struct node
{
int x,y,z;
}edge[maxn];
bool cmp(node a,node b) {
return a.z < b.z;
}
int n,m;
ULL sum;
int vis[maxn];
int Find(int n){
return n==vis[n] ? n : vis[n]=next_get(vis[n]);
}
int Kruskal(){
sort(edge+1,edge+m+1,cmp);
sum=0;
for(int i=1;i<=m;++i){
int x=Find(edge[i].x);
int y=Find(edge[i].y);
if(x==y) continue;
vis[y]=x;
sum+=edge[i].z;
}
}
int main(){
while(~scanf("%d",&n)&&n){
m=n*(n-1)/2;
for(int i=1;i<=m;++i){ scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].z);}
for(int i=0;i<=m;++i){ vis[i]=i;}
Kruskal();
int ans=0;
for(int i=1;i<=n;++i) {if(i==vis[i]) ans++;}
if(ans>1) puts("impossible\n");
else printf("%lld\n",sum);
}
return 0;
}
|