浅谈树上差分

P1 引入

序列上的差分，相当于树是一条链的情况。

$d_i=a_i-a_{i-1}(i\ge 2)$
$d_1=a_1$

那么对于 $[x,y]$ 的区间加：

$d_x\leftarrow d_x+1$
$d_{y+1}\leftarrow d_{y+1}-1$

最终值：

$a_i=\sum _{j=1}^i{d}_j$

	cin >> n >> k;
	while(k -- ) {
		cin >> x >> y;
		d[x] ++ , d[y+1] -- ;
	}
	
	int mx = 0;
	for(int i=1; i<=n; i++)
		a[i] = a[i-1] + d[i],
		mx = max(mx, a[i]);
	cout << mx;

同样的我们也可以用从 $n$ 开始后缀和的形式：

	cin >> n >> k;
	while(k -- ) {
		cin >> x >> y;
		d[y] ++ , d[x-1] --;
	}
	
	int mx = 0;
	for(int i=n; i>=1; i--)
		a[i] = a[i+1] + d[i],
		mx = max(mx, a[i]);
	cout << mx;

由此可见，不同的前缀和方式对应着不同的差分、求具体值方式。

P2 点差分

题目

给定一棵树，多次操作，每次操作在树上一条简单路径上的点都加 $1$，问操作后所有点权最大值。

一般情况中，类比 P1，我们首先需要确定求具体值方式。

记 $a_u$ 表示结点 $u$ 的点权大小。

在树上有两种求结点 $u$ 的具体值方式：

• 自上而下：从根节点开始到 $u$ 求和求具体值。

  这种情况比较容易想到，每个点的差分 $d_u$ 表示的是 $u$ 与其父亲的差值，即 $d_u=a_u-a_{f{a}_u}$。

  这样对于单次路径的修改，我们要将 lca 处的结点 +1，并将所有路径上的点相连的非路径上的点 -1，这样修改的点数是至多 $n$ 个，时间复杂度是$O(n)$ 级别的，不符合要求。

• 自下而上：从叶子结点到结点 $u$ 求和求具体值。

  此时差分表示的是 $u$ 与其所有儿子 $v$ 的和的差值，即 $d_u=a_u-\sum _{v\in so{n}_u}{a}_v$。

  根据求和方式，我们考虑将 $x$ 到 $y$ 的路径拆分为两条链：

  $x\to ...\to \text{lca(x,y)}$ 和 $\text{lca(x,y)}\to ...\to y$ 。（注意不要重复统计 $\text{lca}$ 处）

  这样对于单次路径修改，我们只需修改 $x,y$ 以及 $\text{lca(x,y),fa[lca(x,y)]}$ 即可，时间复杂度 $O(1)$。

核心代码：

void dfs2(int u, int fa)
{
    a[u] = d[u];
    for(auto v : G[u])
    {
        if(v == fa) continue;
        dfs2(v, u);
        a[u] += a[v];
    }
    mx = max(mx, a[u]);
}

int main()
{
    ...
    while(k -- )
    {
        int x, y;
        cin >> x >> y;
        int LCA = lca(x, y);
        
        d[x] ++ , d[LCA] -- ;
        d[y] ++ , d[f[LCA][0]] -- ;
    }
    dfs2(1, 0);
    cout << mx;
    return 0;
}

P3 边差分

给定一棵树，多次操作，每次操作在树上一条简单路径上的边都加 $1$，问操作后所有边权最大值。

考虑转边权为点权：$a_u$ 表示结点 $u$ 与 $f{a}_u$ 间边的权值。

对于在链上的操作 $x\to ...\to so{n}_y\to y$ ，根据 $a$ 的定义，我们应当将 $x$ 到 $so{n}_y$ 之间的点的 $a$ 都加 $1$ ，差分即 $d_x\leftarrow d_x+1,d_y\leftarrow d_y-1$。

对于简单路径 $x$ 到 $y$ 的加操作，即拆分为两条链上路径 $[x,lca)$ 与 $[y,lca)$。

	...
	while(k -- )
    {
        int x, y;
        cin >> x >> y;
        int LCA = lca(x, y);
        d[x] ++ , d[y] ++ , d[LCA] -= 2;
    }
    dfs2(1, 0);
    cout << mx;