BZOJ5293: [Bjoi2018]求和

【传送门：BZOJ5293】

---

简要题意：

　　给出一棵有n个点的树，有m个询问，每个询问输入x,y,k，输出x到y的路径上的所有点的深度的k次方的和

---

题解：

　　因为k很小，直接预处理每个点的k次方深度和k次方的时候根到每个点的深度的k次方的和

　　然后求LCA，直接搞就可以了

---

参考代码：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
struct node
{
    int x,y,next;
}a[700000];int len,last[400000];
void ins(int x,int y)
{
    len++;
    a[len].x=x;a[len].y=y;
    a[len].next=last[x];last[x]=len;
}
LL Mod=998244353;
LL p_mod(LL a,int b)
{
    LL ans=1;
    while(b!=0)
    {
        if(b%2==1) ans=(ans*a)%Mod;
        a=(a*a)%Mod;b/=2;
    }
    return ans;
}
LL dep[400000][55];
int f[400000][21];
int bin[21];
void dfs(int x)
{
    for(int i=1;bin[i]<=dep[x][0];i++)
    {
        f[x][i]=f[f[x][i-1]][i-1];
    }
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=f[x][0])
        {
            f[y][0]=x;
            dep[y][0]=dep[x][0]+1;
            for(int i=1;i<=50;i++) dep[y][i]=(p_mod(dep[y][0],i)+dep[x][i])%Mod;
            dfs(y);
        }
    }
}
int LCA(int x,int y)
{
    if(dep[x][0]<dep[y][0]) swap(x,y);
    for(int i=20;i>=0;i--)
    {
        if(dep[x][0]-dep[y][0]>=bin[i])
        {
            x=f[x][i];
        }
    }
    if(x==y) return x;
    for(int i=20;i>=0;i--)
    {
        if(dep[x][0]>=bin[i]&&f[x][i]!=f[y][i])
        {
            x=f[x][i];y=f[y][i];
        }
    }
    return f[x][0];
}
int main()
{
    bin[0]=1;for(int i=1;i<=20;i++) bin[i]=bin[i-1]<<1;
    int n;
    scanf("%d",&n);
    len=0;memset(last,0,sizeof(last));
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        ins(x,y);ins(y,x);
    }
    for(int i=0;i<=50;i++) dep[1][i]=0LL;
    f[1][0]=0;dfs(1);
    int m;
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        int x,y,k;
        scanf("%d%d%d",&x,&y,&k);
        int lca=LCA(x,y);
        printf("%lld\n",((dep[x][k]+dep[y][k]-dep[lca][k]-dep[f[lca][0]][k]+Mod)%Mod+Mod)%Mod);
    }
    return 0;
}