mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2024-11-23 21:11: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

33
networking_flow/minimum_cut.py
View File

@ -1,10 +1,19 @@
# Minimum cut on Ford_Fulkerson algorithm. # 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): def BFS(graph, s, t, parent):
# Return True if there is node that has not iterated. # Return True if there is node that has not iterated.
visited = [False] * len(graph) visited = [False] * len(graph)
queue=[] queue = [s]
queue.append(s)
visited[s] = True visited[s] = True
while queue: while queue:
@ -17,8 +26,12 @@ def BFS(graph, s, t, parent):
return True if visited[t] else False return True if visited[t] else False
def mincut(graph, source, sink): def mincut(graph, source, sink):
# This array is filled by BFS and to store path """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)) parent = [-1] * (len(graph))
max_flow = 0 max_flow = 0
res = [] res = []
@ -27,7 +40,7 @@ def mincut(graph, source, sink):
path_flow = float("Inf") path_flow = float("Inf")
s = sink s = sink
while(s != source): while s != source:
# Find the minimum value in select path # 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] s = parent[s]
@ -35,7 +48,7 @@ def mincut(graph, source, sink):
max_flow += path_flow max_flow += path_flow
v = sink v = sink
while(v != source): while v != source:
u = parent[v] u = parent[v]
graph[u][v] -= path_flow graph[u][v] -= path_flow
graph[v][u] += path_flow graph[v][u] += path_flow
@ -48,12 +61,6 @@ def mincut(graph, source, sink):
return res 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 if __name__ == "__main__":
print(mincut(graph, source, sink)) print(mincut(test_graph, source=0, sink=5))