p4427 [BJOI2018]求和

Posted 2020-11-13 yzxverygood

分析

我们发现k非常的小

于是我们对于每一个点暴力维护它到根的所有节点的深度k次方之和即可

每次找到lca然后用之前处理出来的数组加加减减即可

代码

#include<bits/stdc++.h>
using namespace std;
const int mod = 998244353;
int pr[300100][20],dep[300100],n,m,tot[300100][52];
vector<int>v[300100];
inline int pw(int x,int p){
    int res=1;
    while(p){
      if(p&1)res=1ll*res*x%mod;
      x=1ll*x*x%mod;
      p>>=1;
    }
    return res;
}
inline void dfs(int x,int fa){
    pr[x][0]=fa;
    dep[x]=dep[fa]+1;
    for(int i=0;i<=50;i++)
      tot[x][i]=(tot[fa][i]+pw(dep[x],i))%mod;
    for(int i=0;i<v[x].size();i++)
      if(v[x][i]!=fa)dfs(v[x][i],x);
}
inline int lca(int x,int y){
    if(dep[x]<dep[y])swap(x,y);
    int len=dep[x]-dep[y];
    for(int i=0;i<=18;i++)
      if((1<<i)&len)x=pr[x][i];
    if(x==y)return x;
    for(int i=18;i>=0;i--)
      if(pr[x][i]!=pr[y][i])
        x=pr[x][i],y=pr[y][i];
    return pr[x][0];
}
int main(){
    int i,j,k;
    scanf("%d",&n);
    for(i=1;i<n;i++){
      int x,y;
      scanf("%d%d",&x,&y);
      v[x].push_back(y);
      v[y].push_back(x);
    }
    dep[0]=-1;
    dfs(1,0);
    for(i=0;i<=50;i++)tot[0][i]=0;
    for(i=1;i<=18;i++)
      for(j=1;j<=n;j++)
        pr[j][i]=pr[pr[j][i-1]][i-1];
    scanf("%d",&m);
    while(m--){
      int x,y;
      scanf("%d%d%d",&x,&y,&k);
      int p=lca(x,y);
      int res=((tot[x][k]-tot[p][k]+tot[y][k]-tot[pr[p][0]][k])%mod+mod)%mod;
      printf("%d
",res);
    }
    return 0;
}