全wa求改
查看原帖
全wa求改
705872
lgy2024楼主2023/1/12 19:01
#include<bits/stdc++.h>
using namespace std;
long long n;
long long cnt,h[300005],pos;
long long w[300005];
long long s[300005];
const long long mod = 998244353;
long long table[300005][50];
long long zz[300005][50];
struct t
{
	long long size,son=-1,dep,top,tid,fa;
} tree[300005];
long long fps[300005];
struct edge
{
	long long to,next;
	long long v;
} e[600005];
void addedge(long long x,long long y)
{
	e[++cnt].to=y;
	e[cnt].next=h[x];
	h[x]=cnt;
}
void dfs(long long x,long long prt,long long d)
{
	tree[x].size=1;
	tree[x].dep=d;
	tree[x].fa=prt;
	for(long long i=h[x]; i; i=e[i].next)
	{
		long long y=e[i].to;
		if(y==prt)
		{
			continue;
		}
		dfs(y,x,d+1);
		tree[x].size+=tree[y].size;
		if(tree[x].son==-1||tree[y].size>tree[tree[x].son].size)
		{
			tree[x].son=y;
		}
	}
}
void dfs2(long long x,long long tpp)
{
	tree[x].tid=++pos;
	fps[pos]=x;
	tree[x].top=tpp;
	if(tree[x].son!=-1)
	{
		dfs2(tree[x].son,tpp);
	}
	for(long long i=h[x]; i; i=e[i].next)
	{
		long long y=e[i].to;
		if(y!=tree[x].fa&&y!=tree[x].son)
		{
			dfs2(y,y);
		}
	}
}
long long asks(long long x,long long y,long long cf)
{
	if(x>y)return 0;
	return (zz[y][cf]-zz[x-1][cf])%mod;
}
long long lca(long long u,long long v)
{
	while(tree[u].top!=tree[v].top)
	{
		if(tree[tree[u].top].dep<tree[tree[v].top].dep)
		{
			swap(u,v);
		}
		u=tree[tree[u].top].fa;
	}
	if(tree[u].dep<tree[v].dep)
	{
		swap(u,v);
	}
	return v;
}
signed main()
{
	cin>>n;
	for(long long i=1; i<n; i++)
	{
		long long x,y;
		cin>>x>>y;
		addedge(x,y);
		addedge(y,x);
	}
	for(long long i=1;i<=50;i++)
	{
		table[1][i]=1;
	}
	for(long long i=2;i<=n;i++)
	{
		for(long long j=1;j<=50;j++)
		{
			table[i][j]=(table[i][j-1]%mod)*(i%mod);
			table[i][j]%=mod;
		}
	}
	for(long long i=1;i<=n;i++)
	{
		for(long long j=1;j<=50;j++)
		{
			zz[i][j]=zz[i-1][j]%mod+table[i][j]%mod;
			zz[i][j]%=mod;
		}
	}
	dfs(1,0,0);
	dfs2(1,1);
	long long q;
	cin>>q;
	for(long long i=1; i<=q; i++)
	{
		long long x,y,k;
		cin>>x>>y>>k;
		int ll=lca(x,y);
		cout<<(asks(tree[ll].dep,tree[x].dep,k)%mod+asks(tree[ll].dep+1,tree[y].dep,k)%mod)%mod<<endl;
	}
	return 0;
}
/*
5
1 2
1 3
2 4
2 5
2
1 1 1
1 1 1
*/
2023/1/12 19:01
加载中...