树上差分和线性差分的最基本思想是一模一样的。
点的差分
我们在线性差分上面是吧线段拆成射线,某一个点的标记代表它及以后的元素全部做一个变化;那我们在树上的某一个结点的标记就代表它所有的父亲(包括自己)都做一个变化。假如我们要在两个节点(x,y)之间的路径上的点全部+1,那么我们只需要在x上做+1的标记,y上做+1的标记,lca(x,y)上做-1的标记,father[lca(x,y)]上做-1的标记,就完成了xy之间路径的变化。最后从叶子结点开始做一个前缀和就可以了。
边的差分
因为边的差分和lca是没有关系的,所以我们只要在x和y上+1,lca上面-2就可以了,比点的差分更简单
这道题题意是:
FJ给他的牛棚的N(2≤N≤50,000)个隔间之间安装了N-1根管道,隔间编号从1到N。所有隔间都被管道连通了。
FJ有K(1≤K≤100,000)条运输牛奶的路线,第i条路线从隔间si运输到隔间ti。一条运输路线会给它的两个端点处的隔间以及中间途径的所有隔间带来一个单位的运输压力,你需要计算压力最大的隔间的压力是多少。
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 2e5 + 10;
int n, k, h[N], cnt, dep[N], f[N][21], vis[N], val[N], ans = -1e9;
struct node {
int v, net, w;
} no[N];
void add(int u, int v, int w) {
no[cnt] = node{v, h[u], w};
h[u] = cnt++;
}
void dfs(int u, int fa) {
dep[u] = dep[fa] + 1;
vis[u] = 1;
for(int i = 1; i <= 20; i++)
f[u][i] = f[f[u][i - 1]][i - 1];
for(int i = h[u]; ~i; i = no[i].net) {
int v = no[i].v;
if(!vis[v])
f[v][0] = u, dfs(v, u);
}
}
int lca(int x, int y) {
if(dep[x] < dep[y])
swap(x, y);
for(int i = 20; i >= 0; i--)
if(dep[f[x][i]] >= dep[y])
x = f[x][i];
if(x == y)
return x;
for(int i = 20; i >= 0; i--)
if(f[x][i] != f[y][i])
x = f[x][i], y = f[y][i];
return f[x][0];
}
void dfss(int u) {
vis[u] = 1;
for(int i = h[u]; ~i; i = no[i].net) {
int v = no[i].v;
if(v != f[u][0]) {
dfss(v);
val[u] += val[v];
}
}
ans = max(ans, val[u]);
}
int main() {
memset(h, -1, sizeof h);
scanf("%d%d", &n, &k);
for(int u, v, i = 1; i < n; i++) {
scanf("%d%d", &u, &v);
add(u, v, 0);
add(v, u, 0);
}
dfs(1, 0);
for(int x, y, i = 1; i <= k; i++) {
scanf("%d%d", &x, &y);
int lc = lca(x, y);
val[x]++, val[y]++, val[lc]--, val[f[lc][0]]--;
}
dfss(1);
cout << ans << endl;
return 0;
}