树上差分（lca）

对边差分

(u,v)上全部加上w，对于差分数组就是：
u加上w，v加上w，lca减去2 × w
用子树中差分数组的和来还原信息
每个点的信息记录的是其到父亲的边的信息

边差分

我们对每条a---b +1
lca(a,b)-1;
统计每个点需要的非树边s
如果 s=0  +m
    s=1   ++
    s>1  +0

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010, M = N * 2;

int n, m;
int h[N], e[M], ne[M], idx;
int depth[N], fa[N][17];
int d[N];
int q[N];
int ans;

void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}

void bfs()
{
    memset(depth, 0x3f, sizeof depth);
    depth[0] = 0, depth[1] = 1;
    int hh = 0, tt = 0;
    q[0] = 1;

    while (hh <= tt)
    {
        int t = q[hh ++ ];
        for (int i = h[t]; ~i; i = ne[i])
        {
            int j = e[i];
            if (depth[j] > depth[t] + 1)
            {
                depth[j] = depth[t] + 1;
                q[ ++ tt] = j;
                fa[j][0] = t;
                for (int k = 1; k <= 16; k ++ )
                    fa[j][k] = fa[fa[j][k - 1]][k - 1];
            }
        }
    }
}

int lca(int a, int b)
{
    if (depth[a] < depth[b]) swap(a, b);
    for (int k = 16; k >= 0; k -- )
        if (depth[fa[a][k]] >= depth[b])
            a = fa[a][k];
    if (a == b) return a;
    for (int k = 16; k >= 0; k -- )
        if (fa[a][k] != fa[b][k])
        {
            a = fa[a][k];
            b = fa[b][k];
        }
    return fa[a][0];
}

int dfs(int u, int father)
{
    int res = d[u];
    for (int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if (j != father)
        {
            int s = dfs(j, u);
            if (s == 0) ans += m;
            else if (s == 1) ans ++ ;
            res += s;
        }
    }

    return res;
}

int main()
{
    scanf("%d%d", &n, &m);
    memset(h, -1, sizeof h);
    for (int i = 0; i < n - 1; i ++ )
    {
        int a, b;
        scanf("%d%d", &a, &b);
        add(a, b), add(b, a);
    }

    bfs();

    for (int i = 0; i < m; i ++ )
    {
        int a, b;
        scanf("%d%d", &a, &b);
        int p = lca(a, b);
        d[a] ++, d[b] ++, d[p] -= 2;
    }
    dfs(1, -1);
    printf("%d\n", ans);

    return 0;
}

对点差分

JLOI2014
(u,v)上全部加上w，对于差分数组就是：
u加上w，v加上w，lca减去w，Fatherlca减去w
同样用子树中差分数组的和来还原信息
#include <bits/stdc++.h>
using  namespace std;
//#define int long long
#define inf  1e8
const int N=3e5+100;
int h[N],e[N],ne[N];
int d[N];
int a[N];
int n;
int sum[N];
int fa[N][21];
int cnt;
int q[N];
void add(int a,int b)
{
	e[cnt]=b,ne[cnt]=h[a],h[a]=cnt++;
}
void bfs()
{
	memset(d,0x3f3f,sizeof(d));
   queue <int>q;
	d[0]=0;	
	d[1]=1;
	q.push(1);
	while(!q.empty())
	{
		int t=q.front();
		q.pop();
		for(int i=h[t];~i;i=ne[i])
		{
			int v=e[i];
			if(d[v]>d[t]+1)
			{
				d[v]=d[t]+1;
				q.push(v);
			
				fa[v][0]=t;
				for(int k=1;k<=19;k++)
				{
					fa[v][k]=fa[fa[v] [k-1]][k-1];
				}
			}
		}
	}
}
void dfs(int u,int last)
{
	d[u]=d[last]+1;
	fa[u][0]=last;
	for(int i=1;i<=19;i++)
	{
		fa[u][i]=fa[fa[u][i-1]][i-1];
	}
	for(int i=h[u];~i;i=ne[i])
	{
		int v=e[i];
		if(v==last)continue;
		dfs(v,u);
	}
}
int lca(int a,int b)
{
	if(d[a]<d[b]) swap(a,b);
	for(int i=19;i>=0;i--)
	{
		if(d[fa[a][i]]>=d[b]) a=fa[a][i];
	}
	if(a==b)return a;
	for(int i=19;i>=0;i--)
	{
		if(fa[a][i]!=fa[b][i])
		{
			a=fa[a][i];
			b=fa[b][i];
		}
	}
	return fa[a][0];
}
void dfs2(int u,int fa)
{
	for(int i=h[u];~i;i=ne[i])
	{
		int v=e[i];
		if(v==fa)continue;
		dfs2(v,u);
		sum[u]+=sum[v];
	}
}
int  main()
{
	cin>>n;
	memset(h,-1,sizeof(h));
	for(int i=1;i<=n;i++)cin>>a[i];
	for(int i=1;i<n;i++)
	{
		int a,b;
		cin>>a>>b;
		add(a,b);
		add(b,a);
	}
   
	bfs();
    //dfs(1,-1);
	for(int i=1;i<n;i++)
	{
		int l=lca(a[i],a[i+1]);
		sum[a[i]]++;
		sum[a[i+1]]++;
		sum[l]--;
		sum[fa[l][0]]--;
	}
	dfs2(1,0);
    
	for(int i=2;i<=n;i++)sum [a[i]]--;
	for(int i=1;i<=n;i++)cout<<sum[i]<<endl;
	return 0;
}