Fix atcoder/ac-library#63

FKbelm · Dec 19, 2020 · ed45c72 · ed45c72
1 parent fb3df2d
commit ed45c72
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/Source/AtCoderLibrary/Graph/MaxFlow.cs b/Source/AtCoderLibrary/Graph/MaxFlow.cs
@@ -268,9 +268,9 @@ TValue Dfs(int v, TValue up)
                     _g[v][iter[v]].Cap = op.Add(_g[v][iter[v]].Cap, d);
                     _g[e.To][e.Rev].Cap = op.Subtract(_g[e.To][e.Rev].Cap, d);
                     res = op.Add(res, d);
-                    if (res.Equals(up)) break;
+                    if (res.Equals(up)) return res;
                 }
-
+                level[v] = _n;
                 return res;
             }
 
@@ -283,12 +283,9 @@ TValue Dfs(int v, TValue up)
                 {
                     iter[i] = 0;
                 }
-                while (op.LessThan(flow, flowLimit))
-                {
-                    var f = Dfs(t, op.Subtract(flowLimit, flow));
-                    if (op.Equals(f, default)) break;
-                    flow = op.Add(flow, f);
-                }
+                var f = Dfs(t, op.Subtract(flowLimit, flow));
+                if (op.Equals(f, default)) break;
+                flow = op.Add(flow, f);
             }
             return flow;
         }