题解 [LNOI2014]LCA 树链剖分+离线差分

题解 [LNOI2014]LCA

题目描述

题目链接

具体做法与心路历程

这道题搁了有一段时间，趁周六晚上没有划水的时间里写了这道题。这道题。。也算得上是套路吧。

具体做法

这种问题每次询问一堆点的贡献一般是把贡献进行某种转换，使其与它到祖先这条链扯上关系。一般是能够变成区间修改和区间查询。

我们看题目要求的是$depth[LC{A}_{u,v}]$，这个可以直接转换成把$u$到根的路径上的权值$+1$,然后查询$v$到根节点上的点权和即可。

这么转换后可以考虑没每次的询问$l,r,z$，把$l,r$的点权加到链上然后查询$z$到根的链上权值和。可以用树链剖分+线段树来维护。

对于每个询问的时间复杂度都为$O(nlog{n}^2)$。总的时间复杂度为$O(nqlog{n}^2)$，接受不了。

看到题目的$l,r$我们可以想到差分，即把答案看成$1$ ~ $l-1$和$1$ ~ $r$的贡献差。

我们可以想到把询问离线下来后差分，然后枚举$i$从$1$ ~ $n$,依次把$i$到根的链上的权值$+1$，同时处理掉差分在$i$点的询问，最后输出答案即可。

$\mathcal{C}ode$

边听音乐边码代码效率奇高，20min打完一遍过~

/*******************************
Author:galaxy yr
LANG:C++
Created Time:2019年11月02日 星期六 21时59分00秒
*******************************/
#include<cstdio>
#include<algorithm>
#include<vector>

using namespace std;

struct IO{
    template<typename T>
    IO & operator>>(T&res)
    {
        T q=1;char ch;
        while((ch=getchar())<'0' or ch>'9')if(ch=='-')q=-q;
        res=(ch^48);
        while((ch=getchar())>='0' and ch<='9') res=(res<<1)+(res<<3)+(ch^48);
        res*=q;
        return *this;
    }
}cin;

struct edge{
    int to,next;
    edge(int a=0,int b=0):to(a),next(b){}
};

struct Query{
    int pos,val,id;
    Query(int a=0,int b=0,int c=0):pos(a),val(b),id(c){}
};

const int maxn=5e4+10;
const int mod=201314;

int n,q,head[maxn],cnt,top[maxn],size[maxn],son[maxn],seg[maxn],f[maxn],depth[maxn],ans[maxn];
edge e[maxn<<1];
vector<Query>V[maxn];

void add(int u,int v)
{
    e[++cnt]=edge(v,head[u]);
    head[u]=cnt;
}

/*{{{线段树*/

namespace SegmentTree{

    int tr[maxn*4],lazy[maxn*4];

    void update(int k)
    {
        tr[k]=tr[k<<1]+tr[k<<1|1];
    }

    void work(int k,int l,int r,int val)
    {
        lazy[k]+=val;
        tr[k]+=(r-l+1)*val;
    }

    void pushdown(int k,int l,int r,int mid)
    {
        if(!lazy[k]) return;
        work(k<<1,l,mid,lazy[k]);
        work(k<<1|1,mid+1,r,lazy[k]);
        lazy[k]=0;
    }

    void modify(int k,int l,int r,int x,int y,int val)
    {
        if(l>=x && r<=y) return work(k,l,r,val);
        if(l>y  ||  r<x) return ;
        int mid=(l+r)>>1;
        pushdown(k,l,r,mid);
        modify(k<<1,l,mid,x,y,val);
        modify(k<<1|1,mid+1,r,x,y,val);
        update(k);
    }

    int query(int k,int l,int r,int x,int y)
    {
        if(l>=x && r<=y) return tr[k];
        if(l>y  ||  r<x) return 0;
        int mid=(l+r)>>1;
        pushdown(k,l,r,mid);
        return (query(k<<1,l,mid,x,y)+query(k<<1|1,mid+1,r,x,y));
    }

};

/*}}}*/

/*{{{树链剖分*/

#define v e[i].to

void dfs(int now)
{
    depth[now]=depth[f[now]]+1;
    size[now]=1;
    for(int i=head[now];i;i=e[i].next)
        if(v!=f[now])
        {
            f[v]=now;
            dfs(v);
            size[now]+=size[v];
            if(size[v]>size[son[now]])
                son[now]=v;
        }
}

void dfs(int now,int tp)
{
    seg[now]=++seg[0];
    top[now]=tp;
    if(son[now])
        dfs(son[now],tp);
    for(int i=head[now];i;i=e[i].next)
        if(v!=f[now] && v!=son[now])
            dfs(v,v);
}

void change(int x,int y,int val)
{
    while(top[x]!=top[y])
    {
        if(depth[top[x]]<depth[top[y]]) swap(x,y);
        SegmentTree::modify(1,1,seg[0],seg[top[x]],seg[x],val);
        x=f[top[x]];
    }
    if(depth[x]>depth[y]) swap(x,y);
    SegmentTree::modify(1,1,seg[0],seg[x],seg[y],val);
}

int ask(int x,int y)
{
    int ans=0;
    while(top[x]!=top[y])
    {
        if(depth[top[x]]<depth[top[y]]) swap(x,y);
        ans=(ans+SegmentTree::query(1,1,seg[0],seg[top[x]],seg[x]))%mod;
        x=f[top[x]];
    }
    if(depth[x]>depth[y]) swap(x,y);
    ans=(ans+SegmentTree::query(1,1,seg[0],seg[x],seg[y]))%mod;
    return ans;
}

#undef v

/*}}}*/

int main()
{
    /*freopen("p4211.in","r",stdin);*/
    /*freopen("p4211.out","w",stdout);*/
    cin>>n>>q;
    int v;
    for(int i=2;i<=n;i++)
    {
        cin>>v;
        add(v+1,i);
    }
    dfs(1);
    dfs(1,1);
    int l,r,z;
    for(int i=1;i<=q;i++)
    {
        cin>>l>>r>>z;
        l++,r++,z++;
        V[l-1].push_back(Query(z,-1,i));
        V[r].push_back(Query(z,1,i));
    }
    int res=0;
    for(int i=1;i<=n;i++)
    {
        change(1,i,1);
        for(int j=0;j<V[i].size();j++)
        {
            res=ask(1,V[i][j].pos);
            ans[V[i][j].id]=(ans[V[i][j].id]+V[i][j].val*res+mod)%mod;
        }
    }
    for(int i=1;i<=q;i++)
        printf("%d\n",ans[i]);
    return 0;
}