mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2024-11-27 15:01:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add doctests to networking_flow/minimum_cut.py (#1126)

Browse Source
This commit is contained in:
Christian Clauss 2019-08-13 11:59:49 +02:00 committed by GitHub
parent c74fd0c9bf
commit dc2b575274
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 32 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
networking_flow/minimum_cut.py
View File

@ -1,12 +1,21 @@
# Minimum cut on Ford_Fulkerson algorithm.
test_graph = [
[0, 16, 13, 0, 0, 0],
[0, 0, 10, 12, 0, 0],
[0, 4, 0, 0, 14, 0],
[0, 0, 9, 0, 0, 20],
[0, 0, 0, 7, 0, 4],
[0, 0, 0, 0, 0, 0],
]
def BFS(graph, s, t, parent):
# Return True if there is node that has not iterated.
visited = [False]*len(graph)
queue=[]
queue.append(s)
visited = [False] * len(graph)
queue = [s]
visited[s] = True
while queue:
u = queue.pop(0)
for ind in range(len(graph[u])):
@ -16,26 +25,30 @@ def BFS(graph, s, t, parent):
parent[ind] = u
return True if visited[t] else False
def mincut(graph, source, sink):
# This array is filled by BFS and to store path
parent = [-1]*(len(graph))
max_flow = 0
"""This array is filled by BFS and to store path
>>> mincut(test_graph, source=0, sink=5)
[(1, 3), (4, 3), (4, 5)]
"""
parent = [-1] * (len(graph))
max_flow = 0
res = []
temp = [i[:] for i in graph] # Record orignial cut, copy.
while BFS(graph, source, sink, parent) :
temp = [i[:] for i in graph] # Record orignial cut, copy.
while BFS(graph, source, sink, parent):
path_flow = float("Inf")
s = sink
while(s != source):
while s != source:
# Find the minimum value in select path
path_flow = min (path_flow, graph[parent[s]][s])
path_flow = min(path_flow, graph[parent[s]][s])
s = parent[s]
max_flow += path_flow
max_flow += path_flow
v = sink
while(v != source):
while v != source:
u = parent[v]
graph[u][v] -= path_flow
graph[v][u] += path_flow
@ -44,16 +57,10 @@ def mincut(graph, source, sink):
for i in range(len(graph)):
for j in range(len(graph[0])):
if graph[i][j] == 0 and temp[i][j] > 0:
res.append((i,j))
res.append((i, j))
return res
graph = [[0, 16, 13, 0, 0, 0],
[0, 0, 10 ,12, 0, 0],
[0, 4, 0, 0, 14, 0],
[0, 0, 9, 0, 0, 20],
[0, 0, 0, 7, 0, 4],
[0, 0, 0, 0, 0, 0]]
source, sink = 0, 5
print(mincut(graph, source, sink))
if __name__ == "__main__":
print(mincut(test_graph, source=0, sink=5))