From 42e20d7daf8f2d555703de9b8d353afff074f5af Mon Sep 17 00:00:00 2001 From: subcrip Date: Sun, 19 May 2024 13:24:11 +0800 Subject: [PATCH] Update graph/edmonds-karp.cc Signed-off-by: subcrip --- graph/edmonds-karp.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/graph/edmonds-karp.cc b/graph/edmonds-karp.cc index 1bb27c0..54bfe0a 100644 --- a/graph/edmonds-karp.cc +++ b/graph/edmonds-karp.cc @@ -11,9 +11,9 @@ struct edmonds_karp { edmonds_karp(int n) : edges(n + 1) {} - void add_edge(int from, int to, ll cap, int mark = 0) { + void add_edge(int from, int to, ll cap, int mark = 0, int mark_rev = 0) { edges[from].push_back({to, cap, 0, make_pair(to, int(edges[to].size())), mark}); - edges[to].push_back({from, 0, 0, make_pair(from, int(edges[from].size() - 1)), mark}); + edges[to].push_back({from, 0, 0, make_pair(from, int(edges[from].size() - 1)), mark_rev}); } ll run(int s, int t) {