拿题解用极限数据对拍100多组没问题 但就是WA 请问是代码出现什么问题导致厌氧了呢 有的话大佬捞捞qwq

传送门 提交记录

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=3e5+500,mod=998244353;
struct edge
{
    int to,next;
}e[N<<1];
int n,q,d[N],head[N],cnt=0;
int f[N][21];ll p[N][55];
inline ll read()
{
    ll x=0; char s;
    do{s=getchar();}while(s<'0'||s>'9');
    do{x=(x<<3)+(x<<1)+s-48;s=getchar();}while(s>='0'&&s<='9');
    return x;
}
inline void write(ll x)
{
    if(x>=10)write(x/10);
    putchar(x%10+48);
}
inline ll res(ll x,ll k)
{
    ll ans=1;
    while(k)
    {
        if(k%2)ans=(ans*x)%mod;
        x=(x*x)%mod;
        k>>=1;
    }
    return ans;
}
inline ll find(ll x)
{
    return res(x,mod-2);
}
inline void add(int u,int v)
{
    e[++cnt]=(edge){v,head[u]};
    head[u]=cnt;
}
inline void dfs(int u,int fa)
{
    d[u]=d[fa]+1;
    f[u][0]=fa;
    for(int i=1;i<=20;++i)
        f[u][i]=f[f[u][i-1]][i-1];
    for(int i=head[u];i;i=e[i].next)
    {
        int v=e[i].to;
        if(v==fa) continue;
        dfs(v,u);
    }
}
inline int lca(int u,int v)
{
    if(d[u]<d[v])swap(u,v);
    while(d[u]!=d[v])
        u=f[u][(int)log2(d[u]-d[v])];
    if(u==v) return u;
    for(int i=20;i>=0;--i)
    {
        if(f[u][i]!=f[v][i])
            u=f[u][i],v=f[v][i];
    }
    return f[u][0];
}
inline ll ask(ll x,ll k)
{
    return p[x][k];
}
int main()
{
    int u,v,fa,k; ll ans;
    memset(d,0,sizeof(d));
    memset(head,0,sizeof(head));
    memset(f,0,sizeof(f));
    memset(p,0,sizeof(p));
    
    n=read();
    for(int i=1;i<=n;++i)
        for(int j=1;j<=50;++j)
            p[i][j]=res(i,j),p[i][j]=(p[i][j]+p[i-1][j])%mod;
    for(int i=1;i<=n-1;++i)
        u=read(),v=read(),add(u,v),add(v,u);
    d[0]=-1; dfs(1,0); cin>>q;
    while(q--)
    {
        u=read();v=read();k=read();
        fa=lca(u,v);
        u=d[u]; v=d[v]; fa=d[fa];
        //cout<<u<<" "<<v<<" "<<fa<<endl;
        ans=ask(u,k)+ask(v,k)-ask(fa,k)-ask(fa-1,k);
        ans=(ans%mod+mod)%mod;
        write(ans); putchar(10);
    }
    return 0;
}

qwq