圖的結構 Python實現

1.每個節點類型設置

# 圖的節點結構
class Node:
    def __init__(self, value):
        self.value = value      # 節點值
        self.come = 0           # 節點入度
        self.out = 0            # 節點出度
        self.nexts = []         # 節點的鄰居節點
        self.edges = []         # 在節點為from的情況下，邊的集合

2.每條邊類型設置

# 圖的邊結構
class Edge:
    def __init__(self, weight, fro, to):
        self.weight = weight        # 邊的權重
        self.fro = fro              # 邊的from節點
        self.to = to                # 邊的to節點

3.圖的結構

# 圖結構
class Graph:
    def __init__(self):
        self.nodes = {}     # 圖的所有節點集合  字典形式：{節點編號：節點}
        self.edges = []     # 圖的邊集合

4.圖的實現

# 生成圖結構
# matrix = [
#   [1,2,3],        ==>   里面分別代表權重, from節點, to節點
#   [...]
# ]
from Graph import Graph
from Node import Node
from Edge import Edge


def createGraph(matrix):
    graph = Graph()
    for edge in matrix:
        weight = edge[0]
        fro = edge[1]
        to = edge[2]
        if fro not in graph.nodes:
            graph.nodes[fro] = Node(fro)
        if to not in graph.nodes:
            graph.nodes[to] = Node(to)
        fromNode = graph.nodes[fro]
        toNode = graph.nodes[to]
        newEdge = Edge(weight, fromNode, toNode)
        fromNode.nexts.append(toNode)
        fromNode.out += 1
        toNode.come += 1
        fromNode.edges.append(newEdge)
        graph.edges.append(newEdge)
    return graph