hello-algo/codes/dart/chapter_graph/graph_adjacency_list.dart

125 lines
2.8 KiB
Dart
Raw Normal View History

/**
* File: graph_adjacency_list.dart
* Created Time: 2023-05-15
* Author: liuyuxin (gvenusleo@gmail.com)
*/
import '../utils/vertex.dart';
/* 基于邻接表实现的无向图类 */
class GraphAdjList {
// 邻接表key顶点value该顶点的所有邻接顶点
Map<Vertex, List<Vertex>> adjList = {};
/* 构造方法 */
GraphAdjList(List<List<Vertex>> edges) {
for (List<Vertex> edge in edges) {
addVertex(edge[0]);
addVertex(edge[1]);
addEdge(edge[0], edge[1]);
}
}
/* 获取顶点数量 */
int size() {
return adjList.length;
}
/* 添加边 */
void addEdge(Vertex vet1, Vertex vet2) {
if (!adjList.containsKey(vet1) ||
!adjList.containsKey(vet2) ||
vet1 == vet2) {
throw ArgumentError;
}
// 添加边 vet1 - vet2
adjList[vet1]!.add(vet2);
adjList[vet2]!.add(vet1);
}
/* 删除边 */
void removeEdge(Vertex vet1, Vertex vet2) {
if (!adjList.containsKey(vet1) ||
!adjList.containsKey(vet2) ||
vet1 == vet2) {
throw ArgumentError;
}
// 删除边 vet1 - vet2
adjList[vet1]!.remove(vet2);
adjList[vet2]!.remove(vet1);
}
/* 添加顶点 */
void addVertex(Vertex vet) {
if (adjList.containsKey(vet)) return;
// 在邻接表中添加一个新链表
adjList[vet] = [];
}
/* 删除顶点 */
void removeVertex(Vertex vet) {
if (!adjList.containsKey(vet)) {
throw ArgumentError;
}
// 在邻接表中删除顶点 vet 对应的链表
adjList.remove(vet);
// 遍历其他顶点的链表,删除所有包含 vet 的边
adjList.forEach((key, value) {
value.remove(vet);
});
}
/* 打印邻接表 */
void printAdjList() {
print("邻接表 =");
adjList.forEach((key, value) {
List<int> tmp = [];
for (Vertex vertex in value) {
tmp.add(vertex.val);
}
print("${key.val}: $tmp,");
});
}
}
/* Driver Code */
void main() {
/* 初始化无向图 */
List<Vertex> v = Vertex.valsToVets([1, 3, 2, 5, 4]);
List<List<Vertex>> edges = [
[v[0], v[1]],
[v[0], v[3]],
[v[1], v[2]],
[v[2], v[3]],
[v[2], v[4]],
[v[3], v[4]],
];
GraphAdjList graph = GraphAdjList(edges);
print("\n初始化后,图为");
graph.printAdjList();
/* 添加边 */
// 顶点 1, 2 即 v[0], v[2]
graph.addEdge(v[0], v[2]);
print("\n添加边 1-2 后,图为");
graph.printAdjList();
/* 删除边 */
// 顶点 1, 3 即 v[0], v[1]
graph.removeEdge(v[0], v[1]);
print("\n删除边 1-3 后,图为");
graph.printAdjList();
/* 添加顶点 */
Vertex v5 = Vertex(6);
graph.addVertex(v5);
print("\n添加顶点 6 后,图为");
graph.printAdjList();
/* 删除顶点 */
// 顶点 3 即 v[1]
graph.removeVertex(v[1]);
print("\n删除顶点 3 后,图为");
graph.printAdjList();
}