problem
给定一棵 n n n 个点的有根树,标号 1 ∼ n 1∼n 1∼n, 1 1 1号节点为根。给定常数 k k k。给定 Q Q Q 个询问,每次询问给定 x , y x,y x,y,求:
∑ i = 1 x depth ( lca ( i , y ) ) k \sum _{i=1}^x\texttt{depth}(\texttt{lca}(i,y))^k i=1∑xdepth(lca(i,y))k
lca ( x , y ) \texttt{lca}(x,y) lca(x,y) 表示节点 x x x 与节点 y y y 在有根树上的最近公共祖先。
depth ( x ) \texttt{depth}(x) depth(x) 表示节点 x x x 的深度,根节点的深度为 1 1 1。
答案对 998244353 998244353 998244353 取模。
数据范围: n , Q ≤ 50000 n,Q≤50000 n,Q≤50000, 1 ≤ k ≤ 1 0 9 1≤k≤10^9 1≤k≤109。
solution
这道题和 LCA 差不多。
有一个地方,就是 d e p k dep^k depk 怎么处理。
其实也很简单,在修改操作时,把对每个点加 1 1 1 改成加 d e p k − ( d e p − 1 ) k dep^k-(dep-1)^k depk−(dep−1)k 就可以了。
时间复杂度 O ( n log 2 n ) O(n\log^2n) O(nlog2n)。
code
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#define N 50005
#define P 998244353
using namespace std;
int n,q,k,tot;
int dep[N],fa[N],son[N],top[N],Size[N],pos[N],idx[N],ans[N],tag[N<<2],sum[N<<2],val[N<<2];
struct query{int id,x,y;}Q[N];
vector<int>e[N];
bool operator<(const query &p,const query &q) {return p.x<q.x;}
int add(int x,int y) {return x+y>=P?x+y-P:x+y;}
int dec(int x,int y) {return x-y< 0?x-y+P:x-y;}
int mul(int x,int y) {return 1ll*x*y%P;}
int power(int a,int b){
int ans=1;
for(;b;b>>=1,a=mul(a,a)) if(b&1) ans=mul(ans,a);
return ans;
}
void dfs1(int x){
Size[x]=1;
for(int &to:e[x]){
fa[to]=x,dep[to]=dep[x]+1;
dfs1(to),Size[to]+=Size[x];
if(Size[to]>Size[son[x]]) son[x]=to;
}
}
void dfs2(int x,int tp){
top[x]=tp,idx[pos[x]=++tot]=x;
if(son[x]) dfs2(son[x],top[x]);
for(int &to:e[x])
if(to!=son[x]) dfs2(to,to);
}
#define mid ((l+r)>>1)
void build(int root,int l,int r){
if(l==r){
val[root]=dec(power(dep[idx[l]],k),power(dep[idx[l]]-1,k));
return;
}
build(root<<1,l,mid),build(root<<1|1,mid+1,r);
val[root]=add(val[root<<1],val[root<<1|1]);
}
void addit(int root,int num){
tag[root]=add(tag[root],num);
sum[root]=add(sum[root],mul(num,val[root]));
}
void pushdown(int root){
if(!tag[root]) return;
addit(root<<1,tag[root]),addit(root<<1|1,tag[root]);
tag[root]=0;
}
void Modify(int root,int l,int r,int x,int y){
if(l>=x&&r<=y){
return addit(root,1);
}
pushdown(root);
if(x<=mid) Modify(root<<1,l,mid,x,y);
if(y> mid) Modify(root<<1|1,mid+1,r,x,y);
sum[root]=add(sum[root<<1],sum[root<<1|1]);
}
int Query(int root,int l,int r,int x,int y){
if(l>=x&&r<=y) return sum[root];
pushdown(root);
if(y<=mid) return Query(root<<1,l,mid,x,y);
if(x> mid) return Query(root<<1|1,mid+1,r,x,y);
return add(Query(root<<1,l,mid,x,y),Query(root<<1|1,mid+1,r,x,y));
}
#undef mid
void Add(int x){
while(top[x]!=1){
Modify(1,1,n,pos[top[x]],pos[x]),x=fa[top[x]];
}
Modify(1,1,n,1,pos[x]);
}
int Ask(int x){
int ans=0;
while(top[x]!=1){
ans=add(ans,Query(1,1,n,pos[top[x]],pos[x])),x=fa[top[x]];
}
return add(ans,Query(1,1,n,1,pos[x]));
}
int main(){
scanf("%d%d%d",&n,&q,&k);
for(int i=2,x;i<=n;++i) scanf("%d",&x),e[x].push_back(i);
dep[1]=1,dfs1(1),dfs2(1,1),build(1,1,n);
for(int i=1,x,y;i<=q;++i){
scanf("%d%d",&x,&y),Q[i]=(query){i,x,y};
}
sort(Q+1,Q+q+1);
int now=0;
for(int i=1;i<=q;++i){
while(now<Q[i].x) Add(++now);
ans[Q[i].id]=Ask(Q[i].y);
}
for(int i=1;i<=q;++i) printf("%d\n",ans[i]);
return 0;
}