Wa 0pts,就是一个最普通的树上前缀和,大佬们读一下题一秒明白。
有人能帮我看看吗》谢谢!
#include<bits/stdc++.h>
#define int long long
#define ll long long
using namespace std;
const int N=6e5+10,M=51,K=22,mod=998244353;
int n,m;
int f[N][M],depth[N],fa[N][K],q[N];
int h[N],ne[N],e[N],idx;
void add(int a,int b){
e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void Bfs(int rt){
memset(depth,0x3f,sizeof(depth));
int hh=0,tt=0;
depth[0]=0;depth[rt]=1;
q[0]=1;
while(hh<=tt){
int t=q[hh++];
for(int i=h[t];~i;i=ne[i]){
int j=e[i];
if(depth[j]>depth[t]+1){
depth[j]=depth[t]+1;
q[++tt]=j;
fa[j][0]=t;
for(int k=1;k<K;k++)
fa[j][k]=fa[fa[j][k-1]][k-1];
}
}
}
}
ll qmi(ll x,ll p){
ll res=1;
while(p){
if(p&1) res=(x*res)%mod;
x=(x*x)%mod;p>>=1;
}
return res;
}
void dfs(int x,int last){
for(int i=h[x];~i;i=ne[i]){
int j=e[i];
if(j==last) continue;
for(int k=1;k<M;k++)
f[j][k]=(f[x][k]+qmi(depth[j]-1,k))%mod;
dfs(j,x);
}
}
int LCA(int a,int b){
if(depth[a]<depth[b]) swap(a,b);
for(int k=K-1;k>=0;k--)
if(depth[fa[a][k]]>=depth[b]) a=fa[a][k];
if(a==b) return a;
for(int k=K-1;k>=0;k--)
if(fa[a][k]!=fa[b][k]){a=fa[a][k];b=fa[b][k];}
return fa[a][0];
}
int get(int x,int y,int k){
int lca=LCA(x,y);
int res=((ll)f[x][k]+f[y][k]%mod-(ll)2*f[lca][k]%mod+qmi((ll)depth[lca],(ll)k))%mod;
return res%mod;
}
signed main(){
memset(h,-1,sizeof(h));
scanf("%lld",&n);
for(int i=1;i<=n-1;i++){
int a,b;
scanf("%lld%lld",&a,&b);
add(a,b);add(b,a);
}
for(int i=0;i<M;i++) f[1][i]=1;
Bfs(1);dfs(1,-1);
// cout<<depth[1]<<endl;
scanf("%lld",&m);
while(m--){
int a,b,k;
scanf("%lld%lld%lld",&a,&b,&k);
printf("%lld\n",get(a,b,k));
}
return 0;
}