「GXOI / GZOI2019」旧词

传送门

---

problem

给定一棵 $n$ 个点的有根树，标号 $1\sim n$，$1$号节点为根。给定常数 $k$。给定 $Q$ 个询问，每次询问给定 $x,y$，求：

$$\sum _{i=1}^x\text{depth}(\text{lca}(i,y){)}^k$$

$\text{lca}(x,y)$ 表示节点 $x$ 与节点 $y$ 在有根树上的最近公共祖先。

$\text{depth}(x)$ 表示节点 $x$ 的深度，根节点的深度为 $1$。

答案对 $998244353$ 取模。

数据范围：$n,Q\le 50000$，$1\le k\le 10^9$。

---

solution

这道题和 LCA 差不多。

有一个地方，就是 $de{p}^k$ 怎么处理。

其实也很简单，在修改操作时，把对每个点加 $1$ 改成加 $de{p}^k-(dep-1{)}^k$ 就可以了。

时间复杂度 $O(n{\log }^{2}n)$。

---

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;
}