【BZOJ5507】[GXOI/GZOI2019]舊詞（樹鏈剖分，線段樹）

【BZOJ5507】[GXOI/GZOI2019]舊詞（樹鏈剖分，線段樹）

題面

BZOJ
洛谷

題解

若是\(k=1\)就是鏈並裸題了。。。
其實\(k>1\)發現仍是能夠用相似鏈並的思想，這個東西本質上就是對於當前的一個\(x\)，考慮對於其餘全部點的貢獻，而他們的\(LCA\)必定是\(x\)到根節點鏈上的一個點。那麼對於某個\(x\)的祖先節點，除了\(x\)所在的子樹內，其餘的全部子樹內的點所有會產生這個點的深度的\(k\)次方的貢獻。\(k=1\)的時候這個東西能夠直接作的緣由是由於\(1\)次方的差分能夠直接相減。換到\(k>1\)不過是額外維護一下要減去多少個深度\(-1\)就好了。
那麼樹剖+線段樹或者\(LCT\)就能夠很容易的解決了。
一開始WA的緣由是由於在Modify更新特定節點的時候直接調用了pushup操做，因此此時線段樹要開8倍。。。。

#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
#define MOD 998244353
#define MAX 50500
inline int read()
{
    int x=0;bool t=false;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=true,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return t?-x:x;
}
int n,Q,K;
struct Line{int v,next;}e[MAX];
int h[MAX],cnt=1;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
int fpow(int a,int b){int s=1;while(b){if(b&1)s=1ll*s*a%MOD;a=1ll*a*a%MOD;b>>=1;}return s;}
int hson[MAX],size[MAX],top[MAX],fa[MAX],dep[MAX],dfn[MAX],tim,ln[MAX];
void dfs1(int u,int ff)
{
    fa[u]=ff;dep[u]=dep[ff]+1;size[u]=1;
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;dfs1(v,u);size[u]+=size[v];
        if(size[hson[u]]<size[v])hson[u]=v;
    }
}
void dfs2(int u,int tp)
{
    top[u]=tp;dfn[u]=++tim;ln[tim]=u;
    if(hson[u])dfs2(hson[u],tp);
    for(int i=h[u];i;i=e[i].next)
        if(e[i].v!=hson[u])dfs2(e[i].v,e[i].v);
}
#define lson (now<<1)
#define rson (now<<1|1)
struct Node{int s,w1,w2,v1,v2;}t[MAX<<3];
void pushup(int now){t[now].s=(0ll+t[lson].s+t[rson].s+1ll*t[now].w1*t[now].v1+1ll*t[now].w2*t[now].v2)%MOD;}
void Build(int now,int l,int r)
{
    if(l==r){t[now].v1=fpow(dep[ln[l]],K);t[now].v2=fpow(dep[ln[l]]-1,K);return;}
    int mid=(l+r)>>1;
    Build(lson,l,mid);Build(rson,mid+1,r);
    t[now].v1=(t[lson].v1+t[rson].v1)%MOD;
    t[now].v2=(t[lson].v2+t[rson].v2)%MOD;
}
void Modify(int now,int l,int r,int L,int R,int w1,int w2)
{
    if(L<=l&&r<=R){t[now].w1=(t[now].w1+w1)%MOD;t[now].w2=(t[now].w2+w2)%MOD;pushup(now);return;}
    int mid=(l+r)>>1;
    if(L<=mid)Modify(lson,l,mid,L,R,w1,w2);
    if(R>mid)Modify(rson,mid+1,r,L,R,w1,w2);
    pushup(now);
}
int Query(int now,int l,int r,int L,int R,int c1,int c2)
{
    if(L==l&&r==R)return (t[now].s+1ll*c1*t[now].v1+1ll*c2*t[now].v2)%MOD;
    int mid=(l+r)>>1;c1=(c1+t[now].w1)%MOD;c2=(c2+t[now].w2)%MOD;
    if(R<=mid)return Query(lson,l,mid,L,R,c1,c2);
    if(L>mid)return Query(rson,mid+1,r,L,R,c1,c2);
    return (Query(lson,l,mid,L,mid,c1,c2)+Query(rson,mid+1,r,mid+1,R,c1,c2))%MOD;
}
int Y[MAX],ans[MAX];vector<int> Qry[MAX];
int main()
{
    n=read();Q=read();K=read();
    for(int i=2;i<=n;++i)Add(read(),i);
    dfs1(1,0);dfs2(1,1);Build(1,1,n);
    for(int i=1,x;i<=Q;++i)x=read(),Y[i]=read(),Qry[x].push_back(i);
    for(int i=1;i<=n;++i)
    {
        int u=i;while(u)Modify(1,1,n,dfn[top[u]],dfn[u],1,(MOD-1)),u=fa[top[u]];
        for(int j=0;j<Qry[i].size();++j)
        {
            int v=Qry[i][j],y=Y[v];
            while(y)ans[v]=(ans[v]+Query(1,1,n,dfn[top[y]],dfn[y],0,0))%MOD,y=fa[top[y]];
        }
    }
    for(int i=1;i<=Q;++i)printf("%d\n",ans[i]);
    return 0;
}