树上差分与LCA(最近公共祖先)【题目与模板】

背景知识：倍增(建议学ST表)，链式前向星，DFS，差分，前缀和，树形结构。

首先，LCA与树上差分均为树上操作，所以学会建树是第一步。
LCA：根据倍增的思想在建树的时候顺带存储节点间的父级关系，再利用存储好的信息来寻找最近公共祖先。
经典倍增法求LCA
https://www.luogu.com.cn/problem/P3379

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct zzz
{
    int t, nex;
}e[500010 << 1];
int head[500010], tot;
void add(int x, int y)//向前星加边
{
	e[++tot].t = y;
	e[tot].nex = head[x];
	head[x] = tot;
}
int depth[500001], fa[500001][22]/*节点i的2^j级祖先*/, lg[500001];
void dfs(int now, int fath)///now表示当前节点，fath表示它的父亲节点
{
	fa[now][0]=fath;
	depth[now]=depth[fath]+1;
	for(int i=1;i<=lg[depth[now]];i++)///在当前节点的深度允许范围内存储祖先
    {
        fa[now][i]=fa[fa[now][i-1]][i-1];///意思是now的2^i祖先等于now的2^(i-1)祖先的2^(i-1)祖先
    }
	for(int i=head[now];i;i=e[i].nex)
    {
        if(e[i].t != fath)///i作为now节点的第一条边，如果去向不是fath，那么继续深搜，如果是fath，说明没有继续向下的节点了便停止搜索。
        {
            dfs(e[i].t, now);
        }
    }
}
int LCA(int x, int y)///求x和y的最早公共祖先
{
	if(depth[x]<depth[y])
        swap(x, y);
        ///不妨让我们假设深度上来讲x>=y，方便将x和y调到同一深度
	while(depth[x] > depth[y])
		x = fa[x][lg[depth[x]-depth[y]]-1];///x等于x与y两点深度差级的祖先，最终调到同一深度
	if(x == y) ///感觉useless
        return x;
	for(int k = lg[depth[x]] - 1; k >= 0; --k)
		if(fa[x][k] != fa[y][k])//二者的depth[x]级祖先不相等就上调
			x = fa[x][k], y = fa[y][k];
	return fa[x][0];
}
int main()
{
	int n, m, s;
	scanf("%d%d%d", &n, &m, &s);
	for(int i = 1,x,y; i <= n-1; ++i)
    {
		cin>>x>>y;
		add(x, y);
		add(y, x);
	}
	for(int i = 1; i <= n; ++i)//预先算出log_2(i)+1的值，用的时候直接调用就可以了
    {
        lg[i] = lg[i-1] + (1 << lg[i-1] == i);
    }
	dfs(s,0);
	for(int i = 1,x,y; i <= m; ++i)
    {
		cin>>x>>y;
		cout<<LCA(x,y)<<endl;
	}
	return 0;
}

树上差分：分为边和点两种操作

裸题情况
边操作：给定n次操作每次把点u到点v的边权加x
此时我们可以令u，v两点点权值加x，并且令LCA(u,v)权值-2x
点操作：给定n次操作，每次把点u到点v的点权加x
此时我们可以令u，v两点点权加x，并且令LCA(u,v)点权-x，再令LCA(u,v)的父节点点权-x

点操作裸题
边操作裸题

树上差分模板（点操作）

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
struct edge
{
    int nex,to;
};
edge e[100000+10];
int head[50000+10],tot;
void add(int from,int to)
{
    e[++tot].to=to;
    e[tot].nex=head[from];
    head[from]=tot;
}
int lg[100000+100],fa[100000][30],dep[1000000+10];
void DFS(int now,int fath)
{
    dep[now]=dep[fath]+1;
    fa[now][0]=fath;
    for(int i=1;i<=lg[dep[now]];i++)
    {
        fa[now][i]=fa[fa[now][i-1]][i-1];
    }
    for(int i=head[now];i;i=e[i].nex)
    {
        if(e[i].to!=fath)
        {
            DFS(e[i].to,now);
        }
    }
}
int lca(int x,int y)
{
    if(dep[y]>dep[x])
    {
        swap(x,y);
    }
    while(dep[x]>dep[y])
    {
        x=fa[x][lg[dep[x]-dep[y]]-1];
    }
    if(x==y)
        return x;
    for(int k=lg[dep[x]]-1;k>=0;k--)
    {
        if(fa[x][k]!=fa[y][k])
        {
            x=fa[x][k];
            y=fa[y][k];
        }
    }
    return  fa[x][0];
}
int power[50000+100],ans;
void get(int now,int fath)
{
    for(int i=head[now];i;i=e[i].nex)
    {
        if(e[i].to==fath)
            continue ;
            get(e[i].to,now);
        power[now]+=power[e[i].to];
    }
    ans=max(ans,power[now]);
}
int main()
{
    freopen("in.txt","r",stdin);
    int n,k;
    cin>>n>>k;
    for(int i=1,x,y;i<=n-1;i++)
    {
        cin>>x>>y;
        add(x,y);
        add(y,x);
    }
    for(int i=1;i<=n;i++)
    {
        lg[i]=lg[i-1]+(1<<lg[i-1]==i);
    }
    DFS(1,0);
    for(int i=1,x,y;i<=k;i++)
    {
        cin>>x>>y;
        int a=lca(x,y);
        power[x]++,power[y]++,power[a]--,power[fa[a][0]]--;
    }
    get(1,0);
    cout<<ans<<endl;
    return 0;
}