This documentation is automatically generated by online-judge-tools/verification-helper
有向グラフの各辺に下限制約 lower と上限制約 upper を持つ
s-t 最大流を求める。
各辺 e=(u,v) の流量 f_e は
lower_e <= f_e <= upper_eを満たす。さらに(s,t 以外で)フロー保存を満たす中で、s から t への流量を最大化する。
max_flow(s, t) : $O(V^2 E)$(Dinic ベース)MaxFlowLowerBound<long long> g(n);add_edge(from, to, lower, upper) で辺を追加auto [ok, ans] = g.max_flow(s, t);ok == false なら制約を満たす流れが存在しないint n : 頂点数void add_edge(int from, int to, T lower, T upper) : 下限付き有向辺を追加pair<bool, T> max_flow(int s, int t) : 実行可能性判定と最大流量を返すss・超終点 tt と t -> s の無限辺を加えた可行循環で可否判定。t -> s 辺を消して残余グラフで s -> t の追加増加分を流し、最大流を得る。#include "../flow/dinic.cpp"
template<class T>
class MaxFlowLowerBound {
struct raw_edge {
int from{}, to{};
T lower{}, upper{};
};
public:
int n;
vector<raw_edge> edges;
MaxFlowLowerBound() = default;
explicit MaxFlowLowerBound(int n) : n(n) {}
void add_edge(int from, int to, T lower, T upper) {
edges.push_back({from, to, lower, upper});
}
pair<bool, T> max_flow(int s, int t) {
int ss = n, tt = n + 1;
Dinic<T, true> mf(n + 2);
vector<T> b(n, 0);
auto add_edge = [&](int from, int to, T cap) {
int idx = (int)mf.G[from].size();
mf.add_edge(from, to, cap);
return pair<int, int>{from, idx};
};
for(auto &&e : edges) {
mf.add_edge(e.from, e.to, e.upper - e.lower);
b[e.from] -= e.lower;
b[e.to] += e.lower;
}
auto ts = add_edge(t, s, INF<T>);
T req = 0;
vector<pair<int, int>> super_edges;
for(int v = 0; v < n; ++v) {
if(b[v] > 0) {
req += b[v];
super_edges.emplace_back(add_edge(ss, v, b[v]));
} else if(b[v] < 0) {
mf.add_edge(v, tt, -b[v]);
}
}
if(mf.flow(ss, tt) != req) return {false, 0};
for(auto &&id : super_edges) {
if(mf.G[id.first][id.second].cap != 0) return {false, 0};
}
int to = mf.G[ts.first][ts.second].to;
int rev = mf.G[ts.first][ts.second].rev;
T base = mf.G[to][rev].cap;
mf.G[ts.first][ts.second].cap = 0;
mf.G[to][rev].cap = 0;
T add = mf.flow(s, t);
return {true, base + add};
}
};
/**
* @brief 下限制約付きs-t最大流 (Max Flow with Lower Bounds)
*/#line 1 "flow/dinic.cpp"
template<class T, bool directed>
class Dinic {
void bfs(int s){
fill(level.begin(),level.end(), -1);
queue<int> Q;
level[s] = 0;
Q.emplace(s);
while(!Q.empty()){
int v = Q.front(); Q.pop();
for (auto &&e : G[v]){
if(e.cap > 0 && level[e.to] < 0){
level[e.to] = level[v] + 1;
Q.emplace(e.to);
}
}
}
}
T dfs(int v, int t, T f){
if(v == t) return f;
for(int &i = iter[v]; i < G[v].size(); i++){
edge &e = G[v][i];
if(e.cap > 0 && level[v] < level[e.to]){
T d = dfs(e.to, t, min(f, e.cap));
if(d == 0) continue;
e.cap -= d;
G[e.to][e.rev].cap += d;
return d;
}
}
return 0;
}
public:
struct edge {
int to{}; T cap; int rev{};
edge() = default;
edge(int to, T cap, int rev) : to(to), cap(cap), rev(rev) {}
};
vector<vector<edge>> G;
vector<int> level, iter;
Dinic() = default;
explicit Dinic(int n) : G(n), level(n), iter(n) {}
void add_edge(int from, int to, T cap){
G[from].emplace_back(to, cap, G[to].size());
G[to].emplace_back(from, directed ? 0 : cap, G[from].size()-1);
}
T flow(int s, int t, T lim = INF<T>){
T ret = 0;
while(true) {
bfs(s);
if(level[t] < 0 || lim == 0) break;
fill(iter.begin(),iter.end(), 0);
while(true){
T f = dfs(s, t, lim);
if(f == 0) break;
ret += f;
lim -= f;
}
}
return ret;
}
};
/**
* @brief Dinic法(Dinic)
*/
#line 2 "graph/maxflow_lower_bound.cpp"
template<class T>
class MaxFlowLowerBound {
struct raw_edge {
int from{}, to{};
T lower{}, upper{};
};
public:
int n;
vector<raw_edge> edges;
MaxFlowLowerBound() = default;
explicit MaxFlowLowerBound(int n) : n(n) {}
void add_edge(int from, int to, T lower, T upper) {
edges.push_back({from, to, lower, upper});
}
pair<bool, T> max_flow(int s, int t) {
int ss = n, tt = n + 1;
Dinic<T, true> mf(n + 2);
vector<T> b(n, 0);
auto add_edge = [&](int from, int to, T cap) {
int idx = (int)mf.G[from].size();
mf.add_edge(from, to, cap);
return pair<int, int>{from, idx};
};
for(auto &&e : edges) {
mf.add_edge(e.from, e.to, e.upper - e.lower);
b[e.from] -= e.lower;
b[e.to] += e.lower;
}
auto ts = add_edge(t, s, INF<T>);
T req = 0;
vector<pair<int, int>> super_edges;
for(int v = 0; v < n; ++v) {
if(b[v] > 0) {
req += b[v];
super_edges.emplace_back(add_edge(ss, v, b[v]));
} else if(b[v] < 0) {
mf.add_edge(v, tt, -b[v]);
}
}
if(mf.flow(ss, tt) != req) return {false, 0};
for(auto &&id : super_edges) {
if(mf.G[id.first][id.second].cap != 0) return {false, 0};
}
int to = mf.G[ts.first][ts.second].to;
int rev = mf.G[ts.first][ts.second].rev;
T base = mf.G[to][rev].cap;
mf.G[ts.first][ts.second].cap = 0;
mf.G[to][rev].cap = 0;
T add = mf.flow(s, t);
return {true, base + add};
}
};
/**
* @brief 下限制約付きs-t最大流 (Max Flow with Lower Bounds)
*/