节点定义
class Node:
def __init__(self, value):
self.value = value
self.come = 0
self.out = 0
self.nexts = []
self.edges = []
边定义
class Edges:
def __init__(self, weight, from, to):
self.weight = weight
self.fro = fro
self.to = to
图结构
class Graph:
def __init__(self):
self.nodes = {}
self.edges = []
图的实现
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
原文:
https://www.cnblogs.com/icekx/p/915244a4.html