[BJOI2018]求和

题目大意：
　　一个$n(nle3 imes10^5)$个结点的有根树。$m(mle3 imes10^5)$次询问，每次询问$x,y$路径上所有结点深度$k(kle50)$次方和。

思路：
　　树上前缀和+LCA即可。

#include<cstdio>
#include<cctype>
#include<algorithm>
#include<sys/mman.h>
#include<sys/stat.h>
typedef long long int64;
class MMapInput {
    private:
        char *buf,*p;
        int size;
    public:
        MMapInput() {
            register int fd=fileno(stdin);
            struct stat sb;
            fstat(fd,&sb);
            size=sb.st_size;
            buf=reinterpret_cast<char*>(mmap(0,size,PROT_READ,MAP_PRIVATE,fileno(stdin),0));
            p=buf;
        }
        char getchar() {
            return (p==buf+size||*p==EOF)?EOF:*p++;
        }
};
MMapInput mmi;
inline int getint() {
    register char ch;
    while(!isdigit(ch=mmi.getchar()));
    register int x=ch^'0';
    while(isdigit(ch=mmi.getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
const int N=3e5+1,logN=19,K=51,mod=998244353;
int h[N];
struct Edge {
    int to,next;
};
Edge e[N<<1];
inline void add_edge(const int &u,const int &v) {
    e[++h[0]]=(Edge){v,h[u]};h[u]=h[0];
    e[++h[0]]=(Edge){u,h[v]};h[v]=h[0];
}
int dep[N]={-1},anc[N][logN],sum[N][K];
inline int log2(const float &x) {
    return ((unsigned&)x>>23&255)-127;
}
void dfs(const int &x,const int &par) {
    dep[x]=dep[anc[x][0]=par]+1;
    for(register int i=1,v=dep[x];i<K;i++) {
        sum[x][i]=(sum[par][i]+v)%mod;
        v=(int64)v*dep[x]%mod;
    }
    for(register int i=1;i<=log2(dep[x]);i++) {
        anc[x][i]=anc[anc[x][i-1]][i-1];
    }
    for(int i=h[x];i;i=e[i].next) {
        const int &y=e[i].to;
        if(y==par) continue;
        dfs(y,x);
    }
}
inline int lca(int x,int y) {
    if(dep[x]!=dep[y]) {
        if(dep[x]<dep[y]) std::swap(x,y);
        for(register int i=log2(dep[x]-dep[y]);~i;i--) {
            if(dep[anc[x][i]]>=dep[y]) x=anc[x][i];
        }
    }
    if(x==y) return x;
    for(register int i=log2(dep[x]);~i;i--) {
        if(anc[x][i]!=anc[y][i]) {
            x=anc[x][i];
            y=anc[y][i];
        }
    }
    return anc[x][0];
}
inline int query(const int &x,const int &y,const int &k) {
    const int z=lca(x,y);
    return ((sum[x][k]+sum[y][k]-sum[z][k]-sum[anc[z][0]][k])%mod+mod)%mod;
}
int main() {
    const int n=getint();
    for(register int i=1;i<n;i++) {
        add_edge(getint(),getint());
    }
    dfs(1,0); 
    for(register int i=getint();i;i--) {
        const int x=getint(),y=getint(),k=getint();
        printf("%d
",query(x,y,k));
    }
    return 0;
}