subcrip
/
cp-templates
1
0
Fork 0
Code

Update graph/mcmf.cc 

Signed-off-by: subcrip <contact@subc.rip>
This commit is contained in:
subcrip 2024-05-19 13:25:55 +08:00
parent c980791c94
commit 182d6283e3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
graph/mcmf.cc
View File

@ -15,9 +15,9 @@ struct mcmf {
mcmf(int n) : edges(n + 1), dis(n + 1), vis(n + 1) {} mcmf(int n) : edges(n + 1), dis(n + 1), vis(n + 1) {}
void add_edge(int from, int to, ll cap, ll cost, int mark = 0) { void add_edge(int from, int to, ll cap, ll cost, int mark = 0, int mark_rev = 0) {
edges[from].push_back({ to, cap, 0, cost, int(edges[to].size()), mark }); edges[from].push_back({ to, cap, 0, cost, int(edges[to].size()), mark });
edges[to].push_back({ from, 0, 0, -cost, int(edges[from].size() - 1), mark }); edges[to].push_back({ from, 0, 0, -cost, int(edges[from].size() - 1), mark_rev });
} }
bool sp(int s, int t) { bool sp(int s, int t) {