树上差分例题

树上差分

P3128
这是我学完lca之后找的一道题，之后发现和lca有个毛线关系，差点下手就是五十行的lct，最后看了看题解就了解了树上差分的思想

思想

我们每一次对于两个点a,b的操作就是把a->lca(a,b)和lca(a,b)->b上的点加上1，我可以用差分的思想。

大家想一下对于一个数列，我把(l,r)的每一个数加上1，是不是就是先把r+1，之后把（l-1)-1，最后求前缀和，就可以求出这个修改过的区间的和，那么树上是不是也是一样的，先把sum[a]++,sum[b]++,之后要想一下sum[lca(a,b)]-=1,sum[fa[lca(a,b)[0]]-=1这里就是说，root->a和root->b被算了两个，所以sum[fa[lca(a,b)[0]]-=1,但是lca也被计算了两次，所以sum[lca(a,b)]-=1，最后dfs统计每一个点的子树的权值和就好了。

对于边权的树上差分其实也差不多，但是在sum更改哪里有不一样

代码实现

#include<bits/stdc++.h>
using namespace std;
const int maxn=50005;
int n,k;//节点数目和管道次数
int head[2*maxn],to[2*maxn],nex[2*maxn],num=0;
int llog[maxn];
void add(int x,int y){to[++num]=y;nex[num]=head[x];head[x]=num;}

int fa[maxn][40],dep[maxn],sum[maxn];
void dfs(int x,int y){
	fa[x][0]=y;
	dep[x]=dep[y]+1;
	for(int i=1;(1<<i)<=dep[x];i++) fa[x][i]=fa[ fa[x][i-1] ][i-1];
	for(int i=head[x];i;i=nex[i]){
		if(to[i]!=y)
			dfs(to[i],x);
	}
}

int lca(int x,int y){
	if(dep[x]<dep[y]) swap(x,y);
	while(dep[x]>dep[y]) x=fa[x][llog[dep[x]-dep[y]]-1];
	if(x==y) return x;
	for(int i=llog[dep[x]]-1;i>=0;i--){
		if(fa[x][i]!=fa[y][i]){
			x=fa[x][i];
			y=fa[y][i];
		}
	}
	return fa[x][0];
}

int ans=0;
void get_ans(int x,int y){
	for(int i=head[x];i;i=nex[i]){
		if(to[i]!=y){
			get_ans(to[i],x);
			sum[x]+=sum[to[i]];
		}
	}
	ans=max(ans,sum[x]);
}


int a,b;

int main(){
	scanf("%d%d",&n,&k);
	for(int i=1;i<n;i++){
		scanf("%d%d",&a,&b);
		add(a,b);
		add(b,a);
	}
	for(int i=1;i<=n;i++){
		llog[i]=llog[i-1]+(1<<llog[i-1]==i);
	}
	dfs(1,0);//求深度
	
	for(int i=1;i<=k;i++){
		scanf("%d%d",&a,&b); 
		sum[a]+=1;
		sum[b]+=1;
		sum[lca(a,b)]-=1;
		sum[fa[lca(a,b)][0]]-=1;
	}//树上差分
	
	get_ans(1,0); //类似于前缀和
	printf("%d",ans);
	return 0;
}