当前位置:首页 > 行业动态 > 正文

python如何构建有向图

在Python中,我们可以使用字典和列表来构建有向图,字典的键表示节点,值表示从该节点出发可以到达的其他节点。

步骤如下:

1、创建一个空字典来存储图。

2、添加节点到图中,如果节点不存在,就创建一个新的键值对,键是节点,值是一个空列表。

3、添加边到图中,如果边的两个节点都存在,就在第一个节点的值列表中添加第二个节点。

代码如下:

创建一个空字典来存储图
graph = {}
添加节点到图中
def add_node(node):
    if node not in graph:
        graph[node] = []
添加边到图中
def add_edge(node1, node2):
    if node1 in graph and node2 in graph:
        graph[node1].append(node2)
添加节点
add_node('A')
add_node('B')
add_node('C')
添加边
add_edge('A', 'B')
add_edge('A', 'C')
print(graph)  # 输出:{'A': ['B', 'C'], 'B': [], 'C': []}

在这个例子中,我们创建了一个有向图,A’节点指向’B’和’C’节点,’B’和’C’节点都没有指向其他节点。

0

随机文章