【树链剖分】【线段树】树的核心

题意

给定一棵带点权的树，初始点权均为0。有两种操作：1.给u子树内所有点权值+1 2.给路径u到v之间的点权值+1 每次修改后输出带权重心 靠近1优先
$n,q\le 10^5$

分析

可以把点权$a_i$的点，拆成$a_i$个点，那么问题就转换成了求重心，树的重心最多有两个，题目要求深度较浅的一个

由于它是重心，在以它为根时，所有子节点子树权值和都小于等于权值总和的一半。又因为它是深度较浅的重心，在以它为根时，1 号点所在的子树权值和一定小于权值总和的一半。（如果等于一半，则原树中 v 的父节点也是重心）

那么也就是说，除去 1 号点所在的子树的剩余部分权值和大于权值总和的一半。 而以 v 为根时 1 的子树其实对应以 1 为根时 v 的子树外。即需要满 足原树中 v 的子树权值和大于权值总和的一半。

此时写出这个树的dfs序列， 因为子树大小大于一半，所以一定经过这个序列的中点，我们可以在树剖后线段树上二分出这个中点对应树上的位置

这个位置一定在答案的子树当中，可以倍增求出最深的满足子树大小大于一半的点即可

树剖不仅仅可以修改路径的问题，对于子树修改同样适用，因为dfs的顺序，一个子树对应在线段树上的编号一定是连续的，只需要维护子树内的最大编号即可

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
int n,q;
vector <int> G[maxn];
int f[maxn][19],val[maxn];
int dep[maxn],son[maxn],siz[maxn];
void dfs(int u,int fa)
{
    siz[u]=1;
    for(auto to:G[u])
    {
        if(to==fa) continue;
        dep[to]=dep[u]+1;
        dfs(to,u);
        siz[u]+=siz[to];
        if(siz[to]>siz[son[u]]) son[u]=to;
    }
}
int top[maxn],rev[maxn],mx[maxn],seg[maxn];
void ddfs(int u,int fa)
{
    top[u]=fa; seg[u]=++seg[0]; rev[seg[0]]=u;
    mx[u]=seg[u];
    if(!son[u]) return;
    ddfs(son[u],fa);
    mx[u]=max(mx[u],mx[son[u]]);
    for(auto to:G[u])
    {
        if(to==f[u][0] || to==son[u]) continue;
        ddfs(to,to);
        mx[u]=max(mx[u],mx[to]);
    }
}
ll tr[maxn<<2],tag[maxn<<2];
void pushdown(int now,int l,int r)
{
    if(!tag[now]) return;
    tag[now<<1]+=tag[now]; tag[now<<1|1]+=tag[now];
    int mid=l+r>>1;
    tr[now<<1]+=(mid-l+1)*tag[now];
    tr[now<<1|1]+=(r-mid)*tag[now];
    tag[now]=0;
}
void modify(int now,int l,int r,int L,int R)
{
    if(l>=L && r<=R)
    {
        tr[now]+=(r-l+1);
        tag[now]++;
        return;
    }
    int mid=l+r>>1;
    pushdown(now,l,r);
    if(L<=mid) modify(now<<1,l,mid,L,R);
    if(mid<R) modify(now<<1|1,mid+1,r,L,R);
    tr[now]=tr[now<<1]+tr[now<<1|1];
}
ll query(int now,int l,int r,int L,int R)
{
    if(l>=L && r<=R) return tr[now];
    int mid=l+r>>1;
    ll res=0;
    pushdown(now,l,r);
    if(L<=mid) res+=query(now<<1,l,mid,L,R);
    if(mid<R) res+=query(now<<1|1,mid+1,r,L,R);
    return res;
}
void change(int x,int y)
{
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		modify(1,1,seg[0],seg[top[x]],seg[x]);
		x=f[top[x]][0];
	}
	if(seg[x]>seg[y]) swap(x,y);
	modify(1,1,seg[0],seg[x],seg[y]);
}
int locate(int now,int l,int r,ll val)
{
    if(l==r) return rev[l];
    int mid=l+r>>1;
    pushdown(now,l,r);
    if(val>tr[now<<1]) return locate(now<<1|1,mid+1,r,val-tr[now<<1]);
    else return locate(now<<1,l,mid,val);
}
int main()
{
    freopen("core.in","r",stdin);
    freopen("core.out","w",stdout);
    scanf("%d%d",&n,&q);
    for(int i=2;i<=n;i++)
    {
        scanf("%d",&f[i][0]);
        G[i].push_back(f[i][0]);
        G[f[i][0]].push_back(i);
    }
    for(int j=1;j<=18;j++)
        for(int i=1;i<=n;i++)
            f[i][j]=f[f[i][j-1]][j-1];
    dfs(1,0);
    ddfs(1,0);
    int op,x,y;
    for(int i=1;i<=q;i++)
    {
        scanf("%d",&op);
        if(op==1)
        {
            scanf("%d",&x);
            modify(1,1,seg[0],seg[x],mx[x]);
        }
        else
        {
            scanf("%d%d",&x,&y);
            change(x,y);
        }
        ll he=query(1,1,seg[0],1,seg[0])+1;
        int ans=locate(1,1,seg[0],he/2);
        for(int j=18;j>=0;j--)
        {
            ll val=query(1,1,seg[0],seg[f[ans][j]],mx[f[ans][j]]);
            if(f[ans][j] && val<(he-1)/2+1)
                ans=f[ans][j];
        }
        if(query(1,1,seg[0],seg[ans],mx[ans])<(he-1)/2+1)
            ans=f[ans][0];
        printf("%d\n",ans);
    }
    return 0;
}