树上差分,顾名思义就是在树上搞差分。有两种典型题型,一种是边差分,一种是点差分。一般与LCA有关。
边差分裸题长这样:
给你一棵树,有n次修改操作,每次把u…v的路径权值加x,最后问从x…y的路径权值和。
模板题
这题是 点差分
#include<bits/stdc++.h>
using namespace std;
inline int rd(){
int x=0,k=1; char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-') k=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){ x=(x<<3)+(x<<1)+ch-'0'; ch=getchar();}
return k*x;
}
typedef long long ll;
const int N=50010;
struct node{
int to,ne;
}a[N<<1];
int n,m,num,cnt,h[N],ol[N],c[N];
int f[N][21],de[N];
inline void add(int x,int y){
a[++num]=(node){y,h[x]};
h[x]=num;
}
void dfs(int u,int fa){
ol[++cnt]=u; de[u]=de[fa]+1;
f[u][0]=fa;
for(int i=1;i<=20;++i)
f[u][i]=f[f[u][i-1]][i-1];
for(int i=h[u];i;i=a[i].ne){
int v=a[i].to;
if(v==fa) continue;
//c[v]=0;
dfs(v,u);
}
}
int lca(int x,int y){
if(de[x]<de[y]) swap(x,y);
int k=de[x]-de[y];
for(int i=0;i<=20;++i)
if(k&(1<<i)) 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 calc(int u)
{
for(int i=h[u];i;i=a[i].ne){
int v=a[i].to;
if(v==f[u][0]) continue;
calc(v);
c[u]+=c[v];
}
}*/
int main()
{
n=rd(); m=rd(); int x,y;
for(int i=1;i<n;++i){
x=rd(); y=rd();
add(x,y); add(y,x);
}
dfs(1,0);
for(int i=1;i<=m;++i){
x=rd(); y=rd();
int an=lca(x,y);
c[x]++; c[y]++;
c[an]--; c[f[an][0]]--;
}
for(int i=n;i>=1;--i) c[f[ol[i]][0]]+=c[ol[i]]; //dfs序累加权值
//calc(1); //或者一遍dfs累加
/*for(int i=1;i<=n;++i)
cout<<c[i]<<" ";*/
int ans=0;
for(int i=1;i<=n;++i)
ans=max(ans,c[i]);
printf("%d\n",ans);
return 0;
}
边差分 运输计划
void dfs(int x,int fa){
p[++id]=x;
for(register int i=h[x];i;i=a[i].ne){
int v=a[i].to;
if(v!=fa){
de[v]=de[x]+1; f[v][0]=x;
dis[v]=dis[x]+a[i].d; w[v]=a[i].d; //边转点
dfs(v,x);
}
}
}
inline bool check(int k){
memset(c,0,sizeof(c));
int mx=0,mxlen=0,cnt=0;
for(register int i=1;i<=m;++i)
if(len[i]>k){
mxlen=max(mxlen,len[i]);
int x=u[i],y=v[i];
++c[x]; ++c[y]; c[lc[i]]-=2; ++cnt; //差分
}
//calc(1);
if(cnt==0) return true;
//if(s[cnt]!=-1) return s[cnt]<=k;
for(register int i=1;i<=n;++i)
c[f[p[i]][0]]+=c[p[i]];
for(register int i=2;i<=n;++i)
if(c[i]==cnt) mx=max(mx,w[i]);
return mxlen-mx<=k;
}