[LNOI2014][BZOJ3626]LCA

本文介绍了一种利用树链剖分算法解决特定树结构查询问题的方法,通过离线处理和巧妙的数据结构应用,实现了高效求解区间内各节点与其LCA的深度之和。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目大意

一棵有n个节点的树(根为1),令deep(x)表示点x的深度(到根距加1)。
q个询问,形如(l,r,z),查询

x=lrdeep(lca(x,z))

答案对201314取模。

1n,q50000


题目分析

求区间和太麻烦,我们差分一下,变成求前缀和,那么每个询问就拆分成两个前缀和询问。
统计lca的深度和看起来还是很棘手,我们换个角度想:如果我们将x到根的路径上所有点权值都赋值为1,那么y到根的路径上的点权和显然就是它们lca的深度。
类似的,我们离线处理,将询问的z挂在节点z上,从1n枚举节点,然后将枚举点到根路径所有点加上1,同时处理关于该点的询问,查询该点到根路径上的点权和,就能得到前缀和了。
使用树链剖分,在O(nlog22n)的时间复杂度解决。


P.S.

根据Werkeytom_FTD的说法,强制在线的话可以使用可持久化线段树解决。


代码实现

#include <iostream>
#include <cstdio>
#include <cctype>

using namespace std;

int read()
{
    int x=0,f=1;
    char ch=getchar();
    while (!isdigit(ch)) f=(ch=='-')?-1:f,ch=getchar();
    while (isdigit(ch)) x=x*10+ch-'0',ch=getchar();
    return x*f;
}

const int P=201314;
const int N=50500;
const int Q=50500;
const int A=Q<<1;

struct D
{
    int u,id;
    bool tp;

    D(int u0=0,int id0=0,bool tp0=0){u=u0,id=id0,tp=tp0;}
}aim[A];

struct segment_tree
{
    int sum[N<<2],tag[N<<2];

    void add(int x,int l,int r,int delta){(sum[x]+=1ll*(r-l+1)*delta%P)%=P,(tag[x]+=delta)%=P;}

    void clear(int x,int l,int r)
    {
        if (tag[x])
        {
            if (l!=r)
            {
                int mid=l+r>>1;
                add(x<<1,l,mid,tag[x]),add(x<<1|1,mid+1,r,tag[x]);
            }
            tag[x]=0;
        }
    }

    void modify(int x,int st,int en,int l,int r,int delta)
    {
        clear(x,l,r);
        if (st==l&&en==r)
        {
            add(x,l,r,delta);
            clear(x,l,r);
            return;
        }
        int mid=l+r>>1;
        if (en<=mid) modify(x<<1,st,en,l,mid,delta);
        else if (mid+1<=st) modify(x<<1|1,st,en,mid+1,r,delta);
        else modify(x<<1,st,mid,l,mid,delta),modify(x<<1|1,mid+1,en,mid+1,r,delta);
        sum[x]=(sum[x<<1]+sum[x<<1|1])%P;
    }

    int query(int x,int st,int en,int l,int r)
    {
        clear(x,l,r);
        if (st==l&&en==r) return sum[x];
        int mid=l+r>>1;
        if (en<=mid) return query(x<<1,st,en,l,mid);
        else if (mid+1<=st) return query(x<<1|1,st,en,mid+1,r);
        else return (query(x<<1,st,mid,l,mid)+query(x<<1|1,mid+1,en,mid+1,r))%P;
    }
}t;

int last[N],tov[N],next[N],DFN[N],fa[N],orig[N],hea[N],size[N],head[N];
int n,q,tot,cnt,idx;
int nxt[A];
int ans[Q];

void insert(int x,int y)
{
    tov[++tot]=y,next[tot]=last[x],last[x]=tot;
}

void hang(int x,D y)
{
    aim[++cnt]=y,nxt[cnt]=head[x],head[x]=cnt;
}

void dfs(int x)
{
    size[x]=1;
    for (int i=last[x],y;i;i=next[i])
        dfs(y=tov[i]),size[x]+=size[y];
}

void build(int x,int og)
{
    DFN[x]=++idx,orig[x]=og;
    for (int i=last[x];i;i=next[i])
        if (!hea[x]||size[hea[x]]<size[tov[i]])
            hea[x]=tov[i];
    if (!hea[x]) return;
    build(hea[x],og);
    for (int i=last[x];i;i=next[i])
        if (tov[i]!=hea[x]) build(tov[i],tov[i]);
}

void fillpath(int x)
{
    for (int y=orig[x];x;y=orig[x=fa[y]])
        t.modify(1,DFN[y],DFN[x],1,idx,1);
}

int sumpath(int x)
{
    int ret=0;
    for (int y=orig[x];x;y=orig[x=fa[y]])
        (ret+=t.query(1,DFN[y],DFN[x],1,idx))%=P;
    return ret;
}

void calc(int x)
{
    for (int i=head[x];i;i=nxt[i])
    {
        D qy=aim[i];
        (((ans[qy.id]+=(qy.tp?1:-1)*sumpath(qy.u))%=P)+=P)%=P;
    }
}

int main()
{
    freopen("lca.in","r",stdin),freopen("lca.out","w",stdout);
    n=read(),q=read();
    for (int i=2;i<=n;i++) fa[i]=read()+1,insert(fa[i],i);
    dfs(1),build(1,1);
    for (int i=1,l,r,z;i<=q;i++)
    {
        l=read()+1,r=read()+1,z=read()+1;
        if (l-1) hang(l-1,D(z,i,0));
        hang(r,D(z,i,1));
    }
    for (int i=1;i<=n;i++) fillpath(i),calc(i);
    for (int i=1;i<=q;i++) printf("%d\n",ans[i]);
    fclose(stdin),fclose(stdout);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值