树上差分的整理（点的树上差分和边的树上差分）

点的树上差分

若经过 u 到 v 的所有点，tmp[u]++,tmp[v]++,tmp[LCA(u,v)]–,tmp[parent[LCA(u,v)][0]]–。

例题：https://www.luogu.com.cn/problem/P3258

#include<bits/stdc++.h>
using namespace std;
struct ss{
    int next,to;
};ss data[600010];
int n,q;
int a[300010],head[600010],deep[300010],p[300010][25],sum[300010];
void change(int &a,int &b)
{
    int t=a;a=b;b=t;
}
void add(int a,int b)
{
    data[++q].to=b;
    data[q].next=head[a];
    head[a]=q;
}
void dfs(int a,int fa)
{
    deep[a]=deep[fa]+1;
    p[a][0]=fa;
    for(int i=1;(1<<i)<=deep[a];i++)
    p[a][i]=p[p[a][i-1]][i-1];
    for(int i=head[a];i;i=data[i].next)
    {
        int v=data[i].to;
        if(v!=fa)
        dfs(v,a);
    }
}
int lca(int a,int b)
{
    if(deep[a]>deep[b]) change(a,b);
    for(int i=20;i>=0;i--)
    {
        if(deep[a]<=deep[b]-(1<<i))
        {
            //printf("a=%d %d %d\n",a,b,i);
            b=p[b][i];
        }
    }
    //printf("%d %d\n",a,b);
    if(a==b) return a;
    for(int i=20;i>=0;i--)
    {
        if(p[a][i]!=p[b][i])
        a=p[a][i],b=p[b][i];
    }
    //printf("a=%d\n",a);
    return p[a][0];
}
void search(int a)
{
    for(int i=head[a];i;i=data[i].next)
    {
        int v=data[i].to;
        if(v==p[a][0]) continue;
        search(v);
        sum[a]+=sum[v];
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    scanf("%d",&a[i]);
    for(int i=1;i<=n-1;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);add(v,u);
    }
    dfs(1,0);
    for(int i=1;i<=n-1;i++)
    {
        int x=a[i],y=a[i+1],LCA=lca(x,y);
        //printf("lca=%d\n",LCA);
        sum[x]++;
        sum[y]++;
        sum[LCA]--;
        sum[p[LCA][0]]--;
    }
    search(1);
    //for(int i=2;i<=n;i++)
    //sum[a[i]]--;

    for(int i=1;i<=n;i++)
    printf("%d\n",sum[i]);
    return 0;
}

---

边的树上差分

若经过 u 到 v 的所有边，tmp[u]++, tmp[v]++, tmp[lac(u, v)]-=2。
统计某条边的经过次数只需要统计它的儿子节点的子树和。

例题：https://www.luogu.com.cn/problem/P2680

#include<bits/stdc++.h>
#define N 300005
using namespace std;
int a[N],b[N],l[N],to[N*2],head[N],nextt[N*2],valuee[N*2],vva[N],f[N][22],he[N],dep[N],p[N],dis[N],d[N];
int m,n;
int read()
{
    int x=0,f=1;
    char s=getchar();
    while(s<'0'||s>'9'){if(s=='-') f=-1;s=getchar();}
    while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
    return x*f;
}
int tot;
void add(int a,int b,int z)
{
    tot++;
    to[tot]=b;
    nextt[tot]=head[a];
    head[a]=tot;
    valuee[tot]=z;
}
int pp=0;
void dfs(int x,int ff)
{
    p[++pp]=x;
    for(int i=head[x];i;i=nextt[i])
    {
        int v=to[i];
        if(v==ff) continue;
        dis[v]=dis[x]+valuee[i];
        dep[v]=dep[x]+1;
        vva[v]=valuee[i];
        f[v][0]=x;
        for(int i=1;i<=20;i++) f[v][i]=f[f[v][i-1]][i-1];
        dfs(v,x);
    }
}

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];
}
int max1=0,maxx=0;
bool check(int mid)
{
    memset(he,0,sizeof(he));
    int cnt=0;
    for(int i=1;i<=m;++i)
    {
        if(d[i]>mid)
        {
            he[a[i]]++;he[b[i]]++;he[l[i]]-=2;
            cnt++;
        }
    }
    for(int i=n;i>=1;--i)
    {
        he[f[p[i]][0]]+=he[p[i]];
        if(vva[p[i]]>=maxx-mid&&he[p[i]]==cnt) return true;
    }
    return false;
}

int gao(int l,int r)
{
    int ans;
    while(l<=r)
    {
        int mid=(l+r)>>1;
        if(check(mid)) ans=mid,r=mid-1;
        else l=mid+1;
    }
    return ans; 
} 
int main()
{
//    freopen("transport.in","r",stdin);
//    freopen("transport.out","w",stdout);
    n=read(),m=read();
    int flagg=0;
    for(int i=1;i<n;++i)
    {
        int x=read(),y=read(),z=read();
        add(x,y,z);
        add(y,x,z);
        max1=max(max1,z);
    }
    dis[1]=0;
    dfs(1,1);
    for(int i=1;i<=m;++i)
    {
        a[i]=read(),b[i]=read();
        l[i]=LCA(a[i],b[i]);
        d[i]=dis[a[i]]+dis[b[i]]-2*dis[l[i]];
        maxx=max(maxx,d[i]);
    }
    printf("%d",gao(maxx-max1,maxx+1));
}
/*
6 3
1 2 1
2 3 4
3 4 6
4 5 2
5 6 7
1 3 
2 5
4 6
*/