add

Author: genkimaru

Diff for: src/.gitignore

@@ -0,0 +1 @@
+**/test*

Diff for: src/graph/test.py

@@ -0,0 +1,23 @@
+class Graph:
+    def __init__(self, num_vertices):
+        self.num_vertices = num_vertices
+        self.graph = [[0 for _ in range(num_vertices)] for _ in range(num_vertices)]
+
+    def add_edge(self, vertex1, vertex2):
+        self.graph[vertex1][vertex2] = 1
+        self.graph[vertex2][vertex1] = 1
+
+    def display_graph(self):
+        for row in self.graph:
+            print(row)
+
+# 创建一个图实例
+g = Graph(4)
+
+# 添加边
+g.add_edge(0, 1)
+g.add_edge(0, 2)
+g.add_edge(1, 3)
+
+# 显示图的邻接矩阵
+g.display_graph()

Diff for: src/graph/邻接矩阵实现图.py

@@ -0,0 +1,33 @@
+class Graph:
+    def __init__(self):
+        self.graph = {}
+
+    def add_vertex(self, vertex):
+        if vertex not in self.graph:
+            self.graph[vertex] = []
+
+    def add_edge(self, vertex1, vertex2):
+        if vertex1 in self.graph and vertex2 in self.graph:
+            self.graph[vertex1].append(vertex2)
+            self.graph[vertex2].append(vertex1)
+
+    def display_graph(self):
+        for vertex in self.graph:
+            print(vertex, "->", " -> ".join(self.graph[vertex]))
+
+# 创建一个图实例
+g = Graph()
+
+# 添加顶点
+g.add_vertex('A')
+g.add_vertex('B')
+g.add_vertex('C')
+g.add_vertex('D')
+
+# 添加边
+g.add_edge('A', 'B')
+g.add_edge('A', 'C')
+g.add_edge('B', 'D')
+
+# 显示图
+g.display_graph()