MaxFlow/main.py

25 lines
507 B
Python
Raw Normal View History

2024-12-17 08:53:26 +00:00
# 示例
from MaxFlow import MaxFlow
if __name__ == "__main__":
vertices = 6
g = MaxFlow(vertices)
# 添加边及其容量
g.add_edge(0, 1, 16)
g.add_edge(0, 2, 13)
g.add_edge(1, 2, 10)
g.add_edge(1, 3, 12)
g.add_edge(2, 1, 4)
g.add_edge(2, 4, 14)
g.add_edge(3, 2, 9)
g.add_edge(3, 5, 20)
g.add_edge(4, 3, 7)
g.add_edge(4, 5, 4)
source = 0 # 源点
sink = 5 # 汇点
max_flow = g.dinic(source, sink)
print("最大流为:", max_flow)