always-cautious/ortools-clone
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

finish to numpify simple max flow

Browse Source
This commit is contained in:
Laurent Perron
2022-09-06 14:48:40 +02:00
parent f69de028ba
commit 06f3057ebc
2 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
ortools/graph/samples/simple_max_flow_program.py
View File

@@ -40,7 +40,7 @@ def main():
# [START constraints]
# Add arcs in bulk.
# note: we could have used add_arc_with_capacity(start, end, capacity)
smf.add_arcs_with_capacity(start_nodes, end_nodes, capacities)
all_arcs = smf.add_arcs_with_capacity(start_nodes, end_nodes, capacities)
# [END constraints]
# [START solve]
@@ -55,10 +55,10 @@ def main():
exit(1)
print('Max flow:', smf.optimal_flow())
print('')
print(' Arc Flow / Capacity')
for i in range(smf.num_arcs()):
print('%1s -> %1s %3s / %3s' %
(smf.tail(i), smf.head(i), smf.flow(i), smf.capacity(i)))
print(' Arc Flow / Capacity')
solution_flows = smf.flows(all_arcs)
for arc, flow, capacity in zip(all_arcs, solution_flows, capacities):
print(f'{smf.tail(arc)} / {smf.head(arc)} {flow:3} / {capacity:3}')
print('Source side min-cut:', smf.get_source_side_min_cut())
print('Sink side min-cut:', smf.get_sink_side_min_cut())
# [END print_solution]