求助求助求助求助求助
查看原帖
求助求助求助求助求助
819516
liusir146楼主2024/11/27 20:21

qwqwqwqwqwqwq谢谢谢谢谢谢

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e6+10,Mod=998244353;
struct Node{
	int next,to;
}a[N];
int n,m,head[N],tot,d[N],f[N][21],dz[N];
int s[N][51],mi[N];

void add(int u,int v){
	a[++tot].to=v;
	a[tot].next=head[u];
	head[u]=tot;
}
void dfs(int u,int fa){
	for(int i=1;(1<<i)<=d[u];i++){
		f[u][i]=f[f[u][i-1]][i-1];
	}
	for(int i=head[u];i;i=a[i].next){
		int v=a[i].to;
		if(v==fa)continue ;
		d[v]=d[u]+1;f[v][0]=u;
		for(int j=1;j<=50;j++){
			mi[j]=mi[j-1]*d[v]%Mod;
		}
		for(int j=1;j<=50;j++){
			s[v][j]=(s[u][j]+mi[j])%Mod;
		} 
		dfs(v,u);
	}
}
int lca(int u,int v){
	if(d[u]>d[v])swap(u,v);
	for(int i=20;i>=0;i--){
		if(d[u]<=d[v]-(1<<i)){
			v=f[v][i];
		}
	}
	if(u==v)return u;
	for(int i=20;i>=0;i--){
		if(f[u][i]==f[v][i])continue ;
		else u=f[u][i],v=f[v][i];
	}
	return f[u][0];
}
signed main(){
	cin>>n;
	for(int i=1;i<=n-1;i++){
		int u,v;
		cin>>u>>v;
		add(u,v);
		add(v,u);	
	}
	mi[0]=1;
	dfs(1,0);
	cin>>m;
	while(m--){
		int u,v,k;
		cin>>u>>v>>k;
		int LCA=lca(u,v);
		cout<<(s[u][k]+s[v][k]-s[LCA][k]+s[f[LCA][0]][k]+Mod)%Mod<<"\n";
	}
	return 0;
}
2024/11/27 20:21
加载中...