mirror of
https://github.com/krahets/hello-algo.git
synced 2026-06-28 00:24:21 +00:00
feat: add the section of Graph Traversal (#367)
* Graph dev * Add the section of Graph Traversal. * Add missing Vertex.java * Add mkdocs.yml * Update numbering * Fix indentation and update array.md
This commit is contained in:
@@ -7,19 +7,13 @@
|
||||
package chapter_graph;
|
||||
|
||||
import java.util.*;
|
||||
|
||||
/* 顶点类 */
|
||||
class Vertex {
|
||||
int val;
|
||||
public Vertex(int val) {
|
||||
this.val = val;
|
||||
}
|
||||
}
|
||||
import include.*;
|
||||
|
||||
/* 基于邻接表实现的无向图类 */
|
||||
class GraphAdjList {
|
||||
// 请注意,vertices 和 adjList 中存储的都是 Vertex 对象
|
||||
Map<Vertex, Set<Vertex>> adjList; // 邻接表(使用哈希表实现)
|
||||
// 邻接表,使用哈希表来代替链表,以提升删除边、删除顶点的效率
|
||||
// 请注意,adjList 中的元素是 Vertex 对象
|
||||
Map<Vertex, List<Vertex>> adjList;
|
||||
|
||||
/* 构造方法 */
|
||||
public GraphAdjList(Vertex[][] edges) {
|
||||
@@ -59,26 +53,26 @@ class GraphAdjList {
|
||||
public void addVertex(Vertex vet) {
|
||||
if (adjList.containsKey(vet))
|
||||
return;
|
||||
// 在邻接表中添加一个新链表(即 HashSet)
|
||||
adjList.put(vet, new HashSet<>());
|
||||
// 在邻接表中添加一个新链表
|
||||
adjList.put(vet, new ArrayList<>());
|
||||
}
|
||||
|
||||
/* 删除顶点 */
|
||||
public void removeVertex(Vertex vet) {
|
||||
if (!adjList.containsKey(vet))
|
||||
throw new IllegalArgumentException();
|
||||
// 在邻接表中删除顶点 vet 对应的链表(即 HashSet)
|
||||
// 在邻接表中删除顶点 vet 对应的链表
|
||||
adjList.remove(vet);
|
||||
// 遍历其它顶点的链表(即 HashSet),删除所有包含 vet 的边
|
||||
for (Set<Vertex> set : adjList.values()) {
|
||||
set.remove(vet);
|
||||
// 遍历其它顶点的链表,删除所有包含 vet 的边
|
||||
for (List<Vertex> list : adjList.values()) {
|
||||
list.remove(vet);
|
||||
}
|
||||
}
|
||||
|
||||
/* 打印邻接表 */
|
||||
public void print() {
|
||||
System.out.println("邻接表 =");
|
||||
for (Map.Entry<Vertex, Set<Vertex>> entry : adjList.entrySet()) {
|
||||
for (Map.Entry<Vertex, List<Vertex>> entry : adjList.entrySet()) {
|
||||
List<Integer> tmp = new ArrayList<>();
|
||||
for (Vertex vertex : entry.getValue())
|
||||
tmp.add(vertex.val);
|
||||
@@ -90,25 +84,21 @@ class GraphAdjList {
|
||||
public class graph_adjacency_list {
|
||||
public static void main(String[] args) {
|
||||
/* 初始化无向图 */
|
||||
Vertex v0 = new Vertex(1),
|
||||
v1 = new Vertex(3),
|
||||
v2 = new Vertex(2),
|
||||
v3 = new Vertex(5),
|
||||
v4 = new Vertex(4);
|
||||
Vertex[][] edges = { { v0, v1 }, { v1, v2 }, { v2, v3 }, { v0, v3 }, { v2, v4 }, { v3, v4 } };
|
||||
Vertex[] v = Vertex.valsToVets(new int[] { 1, 3, 2, 5, 4 });
|
||||
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 = new GraphAdjList(edges);
|
||||
System.out.println("\n初始化后,图为");
|
||||
graph.print();
|
||||
|
||||
/* 添加边 */
|
||||
// 顶点 1, 2 即 v0, v2
|
||||
graph.addEdge(v0, v2);
|
||||
// 顶点 1, 2 即 v[0], v[2]
|
||||
graph.addEdge(v[0], v[2]);
|
||||
System.out.println("\n添加边 1-2 后,图为");
|
||||
graph.print();
|
||||
|
||||
/* 删除边 */
|
||||
// 顶点 1, 3 即 v0, v1
|
||||
graph.removeEdge(v0, v1);
|
||||
// 顶点 1, 3 即 v[0], v[1]
|
||||
graph.removeEdge(v[0], v[1]);
|
||||
System.out.println("\n删除边 1-3 后,图为");
|
||||
graph.print();
|
||||
|
||||
@@ -119,8 +109,8 @@ public class graph_adjacency_list {
|
||||
graph.print();
|
||||
|
||||
/* 删除顶点 */
|
||||
// 顶点 3 即 v1
|
||||
graph.removeVertex(v1);
|
||||
// 顶点 3 即 v[1]
|
||||
graph.removeVertex(v[1]);
|
||||
System.out.println("\n删除顶点 3 后,图为");
|
||||
graph.print();
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user