python实现由邻接矩阵形成有向图的代码
时间: 2023-09-03 22:16:34 浏览: 99
python由邻接矩阵生成无/有向图.py
下面是Python实现由邻接矩阵形成有向图的代码:
```python
class DiGraph:
def __init__(self, adj_matrix):
self.adj_matrix = adj_matrix
self.num_nodes = len(adj_matrix)
def add_edge(self, start, end):
self.adj_matrix[start][end] = 1
def remove_edge(self, start, end):
self.adj_matrix[start][end] = 0
def get_adj_matrix(self):
return self.adj_matrix
def get_neighbors(self, node):
neighbors = []
for i in range(self.num_nodes):
if self.adj_matrix[node][i] == 1:
neighbors.append(i)
return neighbors
```
这里我们定义了一个DiGraph类,该类包含了邻接矩阵的构造函数、添加边的方法、删除边的方法、返回邻接矩阵的方法以及获取节点邻居的方法。通过这些方法,我们可以方便地构造有向图,并对图进行各种操作。
阅读全文