白话数据结构之【最小生成树】
一：基本概念
1：什么是生成树？
对于图G
比如
2：最小生成树
一个无向连通图G=(V,E)，最小生成树就是联结所有顶点的边的权值和最小时的子图T，此时T无回路且连接所有的顶点，所以它必须是棵树。就是将原图的n个顶点用
n-1条边的权值夹起来最小的且连通没有回路的图
二：求法
最小生成树可以用kruskal（克鲁斯卡尔）算法或prim（普里姆）算法求出
Prim（普里姆）算法
MST（Minimum Spanning Tree，最小生成树）问题有两种通用的解法，Prim算法就是其中之一，它是从点的方面考虑构建一颗MST，大致思想是：设图G顶点集合为U，首先任意选择图G中的一点作为起始点a，将该点加入集合V，再从集合U-V中找到另一点b使得点b到V中任意一点的权值最小，此时将b点也加入集合V；以此类推，现在的集合V={a，b}，再从集合U-V中找到另一点c使得点c到V中任意一点的权值最小，此时将c点加入集合V，直至所有顶点全部被加入V，此时就构建出了一颗MST。因为有N个顶点，所以该MST就有N-1条边，每一次向集合V中加入一个点，就意味着找到一条MST的边。
用图示和代码说明：
初始状态：
设置2个数据结构：
lowcost[i]:表示以i为终点的边的最小权值,当lowcost[i]=0说明以i为终点的边的最小权值=0,也就是表示i点加入了MST
mst[i]:表示对应lowcost[i]的起点，即说明边
我们假设V1是起始点，进行初始化（*代表无限大，即无通路）：
lowcost[2]=6，lowcost[3]=1，lowcost[4]=5，lowcost[5]=*，lowcost[6]=*
mst[2]=1，mst[3]=1，mst[4]=1，mst[5]=1，mst[6]=1，（所有点默认起点是V1）
明显看出，以V3为终点的边的权值最小=1，所以边
此时，因为点V3的加入，需要更新lowcost数组和mst数组：
lowcost[2]=5，lowcost[3]=0，lowcost[4]=5，lowcost[5]=6，lowcost[6]=4
mst[2]=3，mst[3]=0，mst[4]=1，mst[5]=3，mst[6]=3
明显看出，以V6为终点的边的权值最小=4，所以边
此时，因为点V6的加入，需要更新lowcost数组和mst数组：
lowcost[2]=5，lowcost[3]=0，lowcost[4]=2，lowcost[5]=6，lowcost[6]=0
mst[2]=3，mst[3]=0，mst[4]=6，mst[5]=3，mst[6]=0
明显看出，以V4为终点的边的权值最小=2，所以边
此时，因为点V4的加入，需要更新lowcost数组和mst数组：
lowcost[2]=5，lowcost[3]=0，lowcost[4]=0，lowcost[5]=6，lowcost[6]=0
mst[2]=3，mst[3]=0，mst[4]=0，mst[5]=3，mst[6]=0
明显看出，以V2为终点的边的权值最小=5，所以边
此时，因为点V2的加入，需要更新lowcost数组和mst数组：
lowcost[2]=0，lowcost[3]=0，lowcost[4]=0，lowcost[5]=3，lowcost[6]=0
mst[2]=0，mst[3]=0，mst[4]=0，mst[5]=2，mst[6]=0
很明显，以V5为终点的边的权值最小=3，所以边
lowcost[2]=0，lowcost[3]=0，lowcost[4]=0，lowcost[5]=0，lowcost[6]=0
mst[2]=0，mst[3]=0，mst[4]=0，mst[5]=0，mst[6]=0
至此，MST构建成功，如图所示：
根据上面的过程，可以容易的写出具体实现代码如下（cpp）：
#include
#include
#define MAX 100
#define MAXCOST 0x7fffffff
/*
测试数据如下：
7 11
A B 7
A D 5
B C 8
B D 9
B E 7
C E 5
D E 15
D F 6
E F 8
E G 9
F G 11
输出
A - D : 5
D - F : 6
A - B : 7
B - E : 7
E - C : 5
E - G : 9
Total:39
*/
int graph[MAX][MAX];
int Prim(int graph[][MAX], int n)
{
/* lowcost[i]记录以i为终点的边的最小权值，当lowcost[i]=0时表示终点i加入生成树 */
int lowcost[MAX];
/* mst[i]记录对应lowcost[i]的起点，当mst[i]=0时表示起点i加入生成树 */
int mst[MAX];
int i, j, min, minid, sum = 0;
/* 默认选择1号节点加入生成树，从2号节点开始初始化 */
for (i = 2; i <= n; i++)
{
/* 最短距离初始化为其他节点到1号节点的距离 */
lowcost[i] = graph[1][i];
/* 标记所有节点的起点皆为默认的1号节点 */
mst[i] = 1;
}
/* 标记1号节点加入生成树 */
mst[1] = 0;
/* n个节点至少需要n-1条边构成最小生成树 */
for (i = 2; i <= n; i++)
{
min = MAXCOST;
minid = 0;
/* 找满足条件的最小权值边的节点minid */
for (j = 2; j <= n; j++)
{
/* 边权值较小且不在生成树中 */
if (lowcost[j] < min && lowcost[j] != 0)
{
min = lowcost[j];
minid = j;
}
}
/* 输出生成树边的信息:起点，终点，权值 */
printf("%c - %c : %d/n", mst[minid] + 'A' - 1, minid + 'A' - 1, min);
/* 累加权值 */
sum += min;
/* 标记节点minid加入生成树 */
lowcost[minid] = 0;
/* 更新当前节点minid到其他节点的权值 */
for (j = 2; j <= n; j++)
{
/* 发现更小的权值 */
if (graph[minid][j] < lowcost[j])
{
/* 更新权值信息 */
lowcost[j] = graph[minid][j];
/* 更新最小权值边的起点 */
mst[j] = minid;//如果不用打出路径的话这个数组是不需要的
}
}
}
/* 返回最小权值和 */
return sum;
}
int main()
{
int i, j, k, m, n;
int x, y, cost;
char chx, chy;
/* 读取节点和边的数目 */
scanf("%d%d", &m, &n);
getchar();
/* 初始化图，所有节点间距离为无穷大 */
for (i = 1; i <= m; i++)
{
for (j = 1; j <= m; j++)
{
graph[i][j] = MAXCOST;
}
}
/* 读取边信息 */
for (k = 0; k < n; k++)
{
scanf("%c %c %d", &chx, &chy, &cost);
getchar();
i = chx - 'A' + 1;
j = chy - 'A' + 1;
graph[i][j] = cost;
graph[j][i] = cost;
}
/* 求解最小生成树 */
cost = Prim(graph, m);
/* 输出最小权值和 */
printf("Total:%d/n", cost);
return 0;
}
下面用poj一道模板题测试
|
Building Roads
Description Farmer John had just acquired several new farms! He wants to connect the farms with roads so that he can travel from any farm to any other farm via a sequence of roads; roads already connect some of the farms. Each of the N (1 ≤ N ≤ 1,000) farms (conveniently numbered 1..N) is represented by a position (Xi, Yi) on the plane (0 ≤ Xi ≤ 1,000,000; 0 ≤ Yi ≤ 1,000,000). Given the preexisting M roads (1 ≤ M ≤ 1,000) as pairs of connected farms, help Farmer John determine the smallest length of additional roads he must build to connect all his farms. Input * Line 1: Two space-separated integers: N and M Output * Line 1: Smallest length of additional roads required to connect all farms, printed without rounding to two decimal places. Be sure to calculate distances as 64-bit floating point numbers. Sample Input 4 1 1 1 3 1 2 3 4 3 1 4 Sample Output 4.00 Source
USACO 2007 December Silver
|
prim算法:
| Memory: 8072K | Time: 188MS | |
| Language: C++ | Result: Accepted |
- #include
- #include
- #include
- #include
- #define INF 0x3f3f3f3f
- using namespace std;
- const int N = 1001;
- double graph[N][N];
- bool visit[N];
- int n,M;
- typedef struct
- {
- double x;
- double y;
- }dian;
- dian m[N];
- double prim()
- {
- memset(visit,0,sizeof(visit));
- double low[1001];
- int pos = 1;
- visit[1] = 1;
- double result = 0;
- for(int i = 2; i <= n; i++)
- {
- low[i] = graph[pos][i];
- }
- for(int i = 0; i < n-1; i++)
- {
- double Min = INF;
- for(int j = 1; j <= n; j++)
- {
- if(!visit[j] && Min > low[j])
- {
- Min = low[j];
- pos = j;
- }
- }
- visit[pos] = 1;
- result += Min;
- for(int i = 1; i <= n; i++)
- {
- if(!visit[i] && low[i] > graph[pos][i])
- {
- low[i] = graph[pos][i];
- }
- }
- }
- return result;
- }
- double dis(dian a,dian b)
- {
- return sqrt((a.x - b.x)*(a.x - b.x) + (a.y - b.y)*(a.y - b.y));
- }
- int main()
- {
- // freopen("in.txt","r",stdin);
- while(cin>>n>>M)
- {
- memset(graph,INF,sizeof(graph));
- for(int i = 1; i <=n;i++)
- {
- cin >> m[i].x>>m[i].y;
- }
- for(int i = 1; i <= n; i++)
- {
- for(int j = i + 1; j <= n; j++)
- {
- graph[i][j] = graph[j][i] = dis(m[i],m[j]);
- }
- }
- for(int i = 0; i < M ; i++)
- {
- int a,b;
- cin>>a>>b;
- graph[a][b] = graph[b][a] = 0;
- }
- printf("%.2lf\n",prim());
- }
- return 0;
- }
kruskal（克鲁斯卡尔）算法
定义：
算法思想：
将每条边按照权值从小到大排序，每次取最小的权值边，如果加上该条边会产生环的话则放弃这条边，继续往下找小的，直到找出n-1条边为止，这个可以利用并查集来实现。
算法演示：
将上面poj那题再用kruskal算法做一遍
#include
#include
#include
#include
#include
using namespace std;
const int N = 1001;
const int E = 1000000;
int n, M;
int cent;
int a[N];
int Count = 0;
typedef struct
{
int x;
int y;
double vaule;
}dian;
dian m[E];
typedef struct
{
double x, y;
}situation;
situation p[N];
double dis(situation a, situation b)
{
return sqrt((a.x - b.x)*(a.x - b.x) + (a.y - b.y) * (a.y - b.y));
}
bool cmp(dian a, dian b)
{
return a.vaule < b.vaule;
}
void init()
{
// cent 这里应该初始化到n
for (int i = 1; i <= n; i++)
{
a[i] = i;
}
}
int Find(int x)
{
while (x != a[x])
{
x = a[x];
}
return x;
}
void Union(int x, int y)
{
// 建议做路径压缩
int fx = Find(x);
int fy = Find(y);
if (fx != fy)
{
a[fx] = fy;
}
}
//kruskal模板
double Kruskal()
{
// init(); 不应该在这里init
sort(m, m + cent, cmp);
double result = 0;
for (int i = 0; i < cent&&Count != n - 1; i++)
{
if (Find(m[i].x) != Find(m[i].y))
{
Union(m[i].x, m[i].y);
result += m[i].vaule;
Count++;
}
}
return result;
}
int main()
{
while (cin >> n >> M)
{
for (int i = 1; i <= n; i++)
{
cin >> p[i].x >> p[i].y;
}
cent = 0;
Count = 0;
for (int i = 1; i <= n; i++)
{
for (int j = i + 1; j <= n; j++)
{
m[cent].x = i;
m[cent].y = j;
m[cent++].vaule = dis(p[i], p[j]);
}
}
// init不应该放在Kruskal里面
init();
for (int i = 1; i <= M; i++)
{
int a, b;
cin >> a >> b;
// 这里还是要检查Find a 和 Find b是不是一样，不然Count会错
if (Find(a) != Find(b)) {
Union(a, b);
Count++;
}
}
printf("%.2f\n", Kruskal());
}
return 0;
}