算法 图7 公路村村通

全部每周作业和视频思考题答案和解析 见 浙江大学 数据结构 思考题+每周练习答案

题目:现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。

输入格式:

输入数据包括城镇数目正整数N(≤1000)和候选道路数目M(≤3N);随后的M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见,城镇从1到N编号。

输出格式:

输出村村通需要的最低成本。如果输入数据不足以保证畅通,则输出−1,表示需要建设更多公路。

输入样例:

6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3

输出样例:

12

解答:

老师说这个题很麻烦。

实际上确实很麻烦(如果不用课程里提供的代码作为辅助的话)。

虽然是个简单的基础题,但还是介绍一下整个系统的结构。

首先先要明确我们的程序除了构建图以外,分为3个部分

是最小堆的建立和使用,用来找最小的边

是考虑到图不知道是稀疏还是稠密,我们使用邻接表以及Kruskal算法构建最小生成树

是使用集合的并集来查看图是否构成环路。

注意题目中有个坑:村子下标是1—N,而不是0开头,所以有些地方在设计的时候需要注意。

课程源代码中有个地方,就是构造图的时候有顶点数据则输入,也是不需要的,删除。

同时,因为这里的堆下标是从0开始,所以循环的时候需要让Child=Parent*2+1

代码如下

#include <iostream>
using namespace std;

#define MaxVertexNum 1000
typedef int Vertex;

// 邻接表存储 - Kruskal最小生成树算法 

//-------------------- 顶点并查集定义 --------------------
typedef Vertex ElementType; // 默认元素可以用非负整数表示 
typedef Vertex SetName;     // 默认用根结点的下标作为集合名称 
typedef ElementType SetType[MaxVertexNum]; // 假设集合元素下标从0开始 							    
typedef int WeightType;       // 边的权值设为整型 
typedef char DataType;        // 顶点存储的数据类型设为字符型 


// 边的定义
typedef struct ENode *PtrToENode;
struct ENode {
	Vertex V1, V2;      // 有向边<V1, V2> 
	WeightType Weight;  // 权重 
};
typedef PtrToENode Edge;

// 邻接点的定义 
typedef struct AdjVNode *PtrToAdjVNode;
struct AdjVNode {
	Vertex AdjV;        // 邻接点下标 
	WeightType Weight;  // 边权重 
	PtrToAdjVNode Next;    // 指向下一个邻接点的指针 
};

/* 顶点表头结点的定义 */
typedef struct Vnode {
	PtrToAdjVNode FirstEdge;	// 边表头指针 
	DataType Data;				// 存顶点的数据 
								// 注意:很多情况下,顶点无数据,此时Data可以不用出现 
} AdjList[MaxVertexNum];		// AdjList是邻接表类型 

// 图结点的定义 
typedef struct GNode *PtrToGNode;
struct GNode {
	int Nv;			// 顶点数 
	int Ne;			// 边数   
	AdjList G;		// 邻接表 
};
typedef PtrToGNode LGraph; // 以邻接表方式存储的图类型 
void InitializeVSet(SetType S, int N)
{ // 初始化并查集 
	ElementType X;
	for (X = 0; X<N; X++) S[X] = -1;
}

void Union(SetType S, SetName Root1, SetName Root2)
{ // 这里默认Root1和Root2是不同集合的根结点 
  // 保证小集合并入大集合 
	if (S[Root2] < S[Root1]) {		// 如果集合2比较大 
		S[Root2] += S[Root1];		// 集合1并入集合2  
		S[Root1] = Root2;
	}
	else {							// 如果集合1比较大 
		S[Root1] += S[Root2];		// 集合2并入集合1  
		S[Root2] = Root1;
	}
}

SetName Find(SetType S, ElementType X)
{ // 默认集合元素全部初始化为-1 
	if (S[X] < 0) // 找到集合的根 
		return X;
	else
		return S[X] = Find(S, S[X]); // 路径压缩 
}

bool CheckCycle(SetType VSet, Vertex V1, Vertex V2)
{ // 检查连接V1和V2的边是否在现有的最小生成树子集中构成回路 
	Vertex Root1, Root2;

	Root1 = Find(VSet, V1); // 得到V1所属的连通集名称 
	Root2 = Find(VSet, V2); // 得到V2所属的连通集名称 

	if (Root1 == Root2)		// 若V1和V2已经连通,则该边不能要 
		return false;
	else { // 否则该边可以被收集,同时将V1和V2并入同一连通集 
		Union(VSet, Root1, Root2);
		return true;
	}
}
//-------------------- 并查集定义结束 --------------------
//-------------------- 边的最小堆定义 --------------------
void PercDown(Edge ESet, int p, int N)
{ // 改编代码4.24的PercDown( MaxHeap H, int p )   
  // 将N个元素的边数组中以ESet[p]为根的子堆调整为关于Weight的最小堆 
	int Parent, Child;
	struct ENode X;

	X = ESet[p]; // 取出根结点存放的值 
	for (Parent = p; (Parent * 2 + 1)<N; Parent = Child) {
		//因为这里的堆下标从0开始,所以需要让Child=Parent*2+1
		Child = Parent * 2 + 1;
		if ((Child != N - 1) && (ESet[Child].Weight>ESet[Child + 1].Weight))
			Child++;  // Child指向左右子结点的较小者 
		if (X.Weight <= ESet[Child].Weight) break; // 找到了合适位置 
		else  // 下滤X 
			ESet[Parent] = ESet[Child];
	}
	ESet[Parent] = X;
}

void InitializeESet(LGraph Graph, Edge ESet)
{ // 将图的边存入数组ESet,并且初始化为最小堆 
	Vertex V;
	PtrToAdjVNode W;
	int ECount;

	// 将图的边存入数组ESet 
	ECount = 0;
	for (V = 0; V<Graph->Nv; V++)
		for (W = Graph->G[V].FirstEdge; W; W = W->Next)
			if (V < W->AdjV) { // 避免重复录入无向图的边,只收V1<V2的边 
				ESet[ECount].V1 = V;
				ESet[ECount].V2 = W->AdjV;
				ESet[ECount++].Weight = W->Weight;
			}
	// 初始化为最小堆 
	for (ECount = Graph->Ne / 2; ECount >= 0; ECount--)
		PercDown(ESet, ECount, Graph->Ne);
}
void Swap(Edge ESet1, Edge ESet2) {
	Vertex temp;
	temp = ESet1->V1;
	ESet1->V1 = ESet2->V1;
	ESet2->V1 = temp;
	temp = ESet1->V2;
	ESet1->V2 = ESet2->V2;
	ESet2->V2 = temp;
	WeightType temp2;
	temp2 = ESet1->Weight;
	ESet1->Weight = ESet2->Weight;
	ESet2->Weight = temp2;
}
int GetEdge(Edge ESet, int CurrentSize)
{ // 给定当前堆的大小CurrentSize,将当前最小边位置弹出并调整堆 

  //将最小边与当前堆的最后一个位置的边交换 
	Swap(&ESet[0], &ESet[CurrentSize - 1]);
	// 将剩下的边继续调整成最小堆 
	PercDown(ESet, 0, CurrentSize - 1);

	return CurrentSize - 1; // 返回最小边所在位置 
}
//-------------------- 最小堆定义结束 --------------------

LGraph CreateGraph(int VertexNum)
{ // 初始化一个有VertexNum个顶点但没有边的图 
	Vertex V;
	LGraph Graph;

	Graph = (LGraph)malloc(sizeof(struct GNode)); // 建立图 
	Graph->Nv = VertexNum;
	Graph->Ne = 0;
	// 初始化邻接表头指针 
	// 注意:这里默认顶点编号从0开始,到(Graph->Nv - 1) 
	for (V = 0; V<Graph->Nv; V++)
		Graph->G[V].FirstEdge = NULL;

	return Graph;
}

void InsertEdge(LGraph Graph, Edge E)
{
	PtrToAdjVNode NewNode;

	// 插入边 <V1, V2> 
	// 为V2建立新的邻接点 
	NewNode = (PtrToAdjVNode)malloc(sizeof(struct AdjVNode));
	NewNode->AdjV = E->V2;
	NewNode->Weight = E->Weight;
	// 将V2插入V1的表头 
	NewNode->Next = Graph->G[E->V1].FirstEdge;
	Graph->G[E->V1].FirstEdge = NewNode;

	// 若是无向图,还要插入边 <V2, V1> 
	// 为V1建立新的邻接点 
	NewNode = (PtrToAdjVNode)malloc(sizeof(struct AdjVNode));
	NewNode->AdjV = E->V1;
	NewNode->Weight = E->Weight;
	// 将V1插入V2的表头 
	NewNode->Next = Graph->G[E->V2].FirstEdge;
	Graph->G[E->V2].FirstEdge = NewNode;
}

int Kruskal(LGraph Graph, LGraph MST)
{ // 将最小生成树保存为邻接表存储的图MST,返回最小权重和 
	WeightType TotalWeight;
	int ECount, NextEdge;
	SetType VSet; // 顶点数组 
	Edge ESet;    // 边数组 

	InitializeVSet(VSet, Graph->Nv); // 初始化顶点并查集 
	ESet = (Edge)malloc(sizeof(struct ENode)*Graph->Ne);
	InitializeESet(Graph, ESet); // 初始化边的最小堆 
	// 创建包含所有顶点但没有边的图。注意用邻接表版本 
	MST = CreateGraph(Graph->Nv);
	TotalWeight = 0; // 初始化权重和 
	ECount = 0;      // 初始化收录的边数 

	NextEdge = Graph->Ne; // 原始边集的规模 
	while (ECount < Graph->Nv - 1) {  // 当收集的边不足以构成树时 
		NextEdge = GetEdge(ESet, NextEdge); // 从边集中得到最小边的位置 
		if (NextEdge < 0) // 边集已空 
			break;
		// 如果该边的加入不构成回路,即两端结点不属于同一连通集 
		if (CheckCycle(VSet, ESet[NextEdge].V1, ESet[NextEdge].V2) == true) {
			// 将该边插入MST 
			InsertEdge(MST, ESet + NextEdge);
			TotalWeight += ESet[NextEdge].Weight; // 累计权重 
			ECount++; // 生成树中边数加1 
		}
	}
	if (ECount < Graph->Nv - 1)
		TotalWeight = -1; // 设置错误标记,表示生成树不存在 

	return TotalWeight;
}

LGraph BuildGraph()
{
	LGraph Graph;
	Edge E;
	Vertex V;
	int Nv, i;

	cin >> Nv;   // 读入顶点个数 
	Graph = CreateGraph(Nv); // 初始化有Nv个顶点但没有边的图 

	cin >> Graph->Ne;   // 读入边数 
	if (Graph->Ne != 0) { // 如果有边 
		E = (Edge)malloc(sizeof(struct ENode)); // 建立边结点 
												// 读入边,格式为"起点 终点 权重",插入邻接矩阵 
		for (i = 0; i<Graph->Ne; i++) {
			cin >> E->V1 >> E->V2 >> E->Weight;
			E->V1--;
			E->V2--;
			// 注意:如果权重不是整型,Weight的读入格式要改 
			InsertEdge(Graph, E);
		}
	}

	// 如果顶点有数据的话,读入数据 
	//for (V = 0; V<Graph->Nv; V++)
	//	cin >> Graph->G[V].Data;

	return Graph;
}

int main(void) {

	LGraph myGraph = BuildGraph();
	LGraph myMst = NULL;
	
	cout << Kruskal(myGraph, myMst);
	system("pause");
	return 0;
}


测试结果:

发布了174 篇原创文章 · 获赞 394 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/tiao_god/article/details/105364242