[GXOI/GZOI2019]舊詞

Description

給定一棵 \(n\) 個點的有根樹,節點標號 \(1 \sim n\)\(1\) 號節點爲根。ios

給定常數 \(k\)
給定 \(Q\) 個詢問,每次詢問給定 \(x,y\)git

求:
\[ \sum\limits_{i \le x} \text{depth}(\text{lca}(i,y))^k \]spa

\(\text{lca}(x,y)\) 表示節點 \(x\) 與節點 \(y\) 在有根樹上的最近公共祖先。
\(\text{depth}(x)\) 表示節點 \(x\) 的深度,根節點的深度爲 \(1\)
因爲答案可能很大,你只須要輸出答案模 \(998244353\) 的結果。code

Solution

\(k=1\) 的時候就是 [LNOI2014]LCA.ip

\(k\not =1\) 的時候其實本質上是同樣的,咱們須要保證加入一個節點 \(x\) 的對 \(y\) 的貢獻爲 \(depth(lca(x,y))^k\) ,那麼咱們能夠構造一個序列,把 \(x\) 到根路徑上所有加上這段序列後查詢 y 到根的路徑和會新增 \(depth(lca(x,y))^k\),因此令這個序列爲 \(A\),則 \(A_i=dep(i)^k-dep(i-1)^k\),因爲路徑深度是連續的,因此巧妙地差分掉了,其他的作法和上面那道題類似,再也不贅述。get

#include <iostream>
#include <vector>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <fstream>

using namespace std;

#define LL long long
#define SZ(x) ((int)x.size())
#define ALL(x) (x).begin(), (x).end()
#define MP(x, y) std::make_pair(x, y)
#define DE(x) cout << #x << " = " << x << endl
#define DEBUG(...) fprintf(stderr, __VA_ARGS__)
#define GO cerr << "GO" << endl;

inline void proc_status()
{
    ifstream t("/proc/self/status");
    cerr << string(istreambuf_iterator<char>(t), istreambuf_iterator<char>()) << endl;
}
template<class T> inline T read() 
{
    register char c;
    register T x(0), f(1);
    while (!isdigit(c = getchar())) if (c == '-') f = -1;
    while (x = (x << 1) + (x << 3) + (c xor 48), isdigit(c = getchar()));
    return x * f;
}
template<typename T> inline bool chkmin(T &a, T b) { return a > b ? a = b, 1 : 0; }
template<typename T> inline bool chkmax(T &a, T b) { return a < b ? a = b, 1 : 0; }

const int maxN=5e4;

#define mod (998244353)
inline void Mod(int& x) { x>=mod?x-=mod:0;}

LL qpow(LL a,LL b)
{
    LL ans(1);
    while(b)
    {
        if(b&1)ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}

int n,Q,k;
int ver[maxN<<1],nxt[maxN<<1],head[maxN+2],tot;
int A[maxN+2],dfn[maxN+2],son[maxN+2],size[maxN+2],dep[maxN+2],rev[maxN+2],dfst,top[maxN+2],fa[maxN+2];
int sum[maxN<<2],tag[maxN<<2];
void link(int u,int v)
{
    ver[++tot]=v,nxt[tot]=head[u],head[u]=tot;
}
void DFS1(int u,int f)
{
    fa[u]=f;
    dep[u]=dep[f]+1;
    size[u]=1;
    for(int i=head[u];i;i=nxt[i])
    {
        int v=ver[i];
        if(v==f)continue;
        DFS1(v,u);
        size[u]+=size[v];
        if(size[son[u]]<size[v])son[u]=v;
    }
}
void DFS2(int u,int topf)
{
    dfn[u]=++dfst;
    rev[dfst]=u;
    top[u]=topf;
    if(son[u])DFS2(son[u],topf);
    for(int i=head[u];i;i=nxt[i])
    {
        int v=ver[i];
        if(v==son[u]||v==fa[u])continue;
        DFS2(v,v);
    }
}
void init()
{
    DFS1(1,0);
    DFS2(1,1);
    for(int i=1;i<=n;++i)
        A[i]=(A[i-1]+qpow(dep[rev[i]],k)-qpow(dep[rev[i]]-1,k)+mod)%mod;
}
void push(int x,int v,int l,int r)
{
    tag[x]+=v;
    sum[x]+=(LL)(A[r]-A[l-1]+mod)*v%mod;
    Mod(sum[x]);
}
void pushdown(int x,int l,int r)
{
    if(tag[x])
    {
        int mid((l+r)>>1);
        push(x<<1,tag[x],l,mid);
        push(x<<1|1,tag[x],mid+1,r);
        tag[x]=0;
    }
}
void add(int x,int l,int r,int L,int R)
{
    if(L<=l&&r<=R)
        return push(x,1,l,r);
    int mid=(l+r)>>1;
    pushdown(x,l,r);
    if(L<=mid)add(x<<1,l,mid,L,R);
    if(mid<R)add(x<<1|1,mid+1,r,L,R);
    sum[x]=sum[x<<1]+sum[x<<1|1],Mod(sum[x]);
}
int query(int x,int l,int r,int L,int R)
{
    if(L<=l&&r<=R)
        return sum[x];
    int mid=(l+r)>>1;
    pushdown(x,l,r);
    int ans(0);
    if(L<=mid)ans+=query(x<<1,l,mid,L,R);
    if(mid<R)ans+=query(x<<1|1,mid+1,r,L,R),Mod(ans);
    return ans;
}
vector<pair<int, int> >q[maxN+2];
int ans[maxN+2];
int main() 
{
#ifndef ONLINE_JUDGE
    freopen("xhc.in", "r", stdin);
    freopen("xhc.out", "w", stdout);
#endif
    n=read<int>(),Q=read<int>(),k=read<int>();
    for(int i=2;i<=n;++i)
    {
        int f(read<int>());
        link(f,i);
        link(i,f);
    }
    init();
    for(int i=1;i<=Q;++i)
    {
        int x(read<int>()),y(read<int>());
        q[x].push_back(MP(y,i));
    }
    for(int i=1;i<=n;++i)
    {
        int u=i;
        while(u)
        {
            add(1,1,n,dfn[top[u]],dfn[u]);
            u=fa[top[u]];
        }
        for(int j=0;j<SZ(q[i]);++j)
        {
            int u=q[i][j].first,id=q[i][j].second,res=0;
            while(u)
            {
                res+=query(1,1,n,dfn[top[u]],dfn[u]);
                Mod(res);
                u=fa[top[u]];
            }
            ans[id]=res;
        }
    }
    for(int i=1;i<=Q;++i)
        printf("%d\n",ans[i]);
    return 0;
}
相關文章
相關標籤/搜索