[Codeforces 903G]Yet Another Maxflow Problem（线段树）

Address

• 洛谷 RemoteJudge
• Codeforces 903G

Meaning

• 一张图， $2n$ 个点，分成两部 $A$ 和 $B$ ，每部 $n$ 个点
• 其中 $A_i$ 到 $A_{i+1}$ 有容量为 $x_i$ 的边（ $1\le i<n$ ）
• $B_i$ 到 $B_{i+1}$ 有容量为 $y_i$ 的边（ $1\le i<n$ ）
• 有 $m$ 个三元组 $(x,y,z)$ 表示 $A_x$ 到 $B_y$ 有容量为 $z$ 的边
• 求 $A_1$ 到 $B_n$ 的最大流
• 之后有 $q$ 个操作，每个操作给出两个参数 $v$ 和 $w$ ，把 $x_v$ （ $A_v$ 到 $A_{v+1}$ 的边的容量）修改成 $w$
• 每次修改之后需要重新求出原图的最大流
• $2\le n,m\le 2\times 10^5$
• $0\le q\le 2\times 10^5$
• $1\le x_i,y_i,z,w\le 10^9$

Solution

• 注：下面一律 $x_n=y_0=0$ ， $cap(u,v)$ 为 $u$ 到 $v$ 的边的容量
• 一道比较不错的数据结构题
• 首先最大流等于最小割
• 然后我们考虑这个特殊的图，最小割是否有特殊性
• 很容易得出，最小割一定是
• （1）在 $A$ 部中割掉一条 $A_i$ 到 $A_{i+1}$ 的边（在 $A$ 部也可以什么边都不割，这时视为割边 $(A_n,A_{n+1})$ ）
• （2）在 $B$ 部中割掉一条 $B_{j-1}$ 到 $B_j$ 的边（同样地可以在 $B$ 部中不割任何边，但这时视为割边 $(B_0,B_1)$ ）
• （3）在上面两个的基础上，把所有满足 $u\le i,v\ge j$ 的边 $(A_u,B_v)$ 全部割掉
• 我们看到（3）之后可以得出一个思路，设
• $$f[i]=\underset{j=1}{\overset{n}{\min }}(y_{j-1}+\sum _{u=1}^i\sum _{v=j}^{n}cap(A_u,B_v))$$
• $f[i]$ 的意义即是在钦定割边 $(A_i,A_{i+1})$ 的情况下的最小割（不包括边 $(A_i,A_{i+1})$ ）
• 如果我们能够求出 $f[1\dots n]$ ，那么我们的问题就能解决了
• 最大流（最小割）就是
• $$\underset{i=1}{\overset{n}{\min }}(f[i]+x_i)$$
• 线段树维护 $f[i]+x_i$ 即可
• 然后我们考虑如何求 $f[1\dots n]$
• 考虑从 $1$ 到 $n$ 按顺序枚举 $i$
• 开一棵线段树
• $j$ 位置的叶子节点储存的是 $y_{j-1}+{\sum }_{u=1}^i{\sum }_{v=j}^{n}cap(A_u,B_v)$ ，意义是钦定割边 $(A_i,A_{i+1})$ 和 $(B_{j-1},B_j)$ 之后还需要割掉多大的容量，需要割掉的边容量之和再加上 $cap(B_{j-1},B_j)$ 即 $y_j$
• 线段树维护区间最小值
• 每次 $i$ 从 $i-1$ 加一过来时，枚举边 $(A_i,B_j)$ ，并在线段树上进行一次区间修改，区间 $[1,j]$ 加上 $cap(A_i,B_j)$ ，这时 $f[i]$ 为线段树全局最小值
• 注意一开始叶子 $j$ 存的值是 $y_{j-1}$
• 复杂度 $O(n\log n)$

Code

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define p2 p << 1
#define p3 p << 1 | 1

inline int read()
{
	int res = 0; bool bo = 0; char c;
	while (((c = getchar()) < '0' || c > '9') && c != '-');
	if (c == '-') bo = 1; else res = c - 48;
	while ((c = getchar()) >= '0' && c <= '9')
		res = (res << 3) + (res << 1) + (c - 48);
	return bo ? ~res + 1 : res;
}

template <class T>
inline T Min(const T &a, const T &b) {return a < b ? a : b;}

typedef long long ll;

const int N = 2e5 + 5, M = N << 2;

int n, m, q, a[N], ecnt, nxt[N], adj[N], go[N], val[N];
ll T[M], add[M], b[N], ioi[N];

void add_edge(int u, int v, int w)
{
	nxt[++ecnt] = adj[u]; adj[u] = ecnt; go[ecnt] = v; val[ecnt] = w;
}

void build(int l, int r, int p)
{
	add[p] = 0;
	if (l == r) return (void) (T[p] = b[l]);
	int mid = l + r >> 1;
	build(l, mid, p2); build(mid + 1, r, p3);
	T[p] = Min(T[p2], T[p3]);
}

void down(int p)
{
	add[p2] += add[p]; add[p3] += add[p];
	add[p] = 0;
}

void upt(int p)
{
	T[p] = Min(T[p2] + add[p2], T[p3] + add[p3]);
}

void change(int l, int r, int s, int e, int v, int p)
{
	if (l == s && r == e) return (void) (add[p] += v);
	int mid = l + r >> 1; down(p);
	if (e <= mid) change(l, mid, s, e, v, p2);
	else if (s >= mid + 1) change(mid + 1, r, s, e, v, p3);
	else change(l, mid, s, mid, v, p2),
		change(mid + 1, r, mid + 1, e, v, p3);
	upt(p);
}

int main()
{
	int x, y, z;
	n = read(); m = read(); q = read();
	for (int i = 1; i < n; i++)
		a[i] = read(), b[i + 1] = read();
	while (m--) x = read(), y = read(), z = read(),
		add_edge(x, y, z);
	build(1, n, 1);
	for (int i = 1; i <= n; i++)
	{
		for (int e = adj[i], v = go[e]; e; e = nxt[e], v = go[e])
			change(1, n, 1, v, val[e], 1);
		ioi[i] = T[1] + add[1];
	}
	for (int i = 1; i <= n; i++) b[i] = a[i] + ioi[i];
	build(1, n, 1);
	std::cout << T[1] + add[1] << std::endl;
	while (q--)
	{
		x = read(); y = read();
		change(1, n, x, x, y - a[x], 1);
		a[x] = y;
		printf("%I64d\n", T[1] + add[1]);
	}
	return 0;
}