WA求助
查看原帖
WA求助
315205
Kniqht楼主2023/1/12 13:02

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;
}  

2023/1/12 13:02
加载中...