牛客练习赛D（换根dp）

换根dp。

主要是当x和y中的祖先节点计数时，需要得到以该点为根的信息。

 然后我们要计算 fx+1 的乘积，并且不能计算当前k的值。不能先乘再除，我们就需要前缀积和后缀积。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

#define M 998244353
int i,j,k,n,m,t,x[500500],y[500500];
ll f[500500],g[500500],rx[500500],ry[500500];
vector<pair<int,int> >v[500500];

ll su(ll a,ll b){a+=b;return (a>=M)?a-M:a;}

void dfs(int z,int fa){
	f[z]=1;
	for(auto [i,j]:v[z]){
		if(i==fa)continue;
		dfs(i,z);
		f[z]=f[z]*(f[i]+1)%M;
		g[z]=su(g[z],g[i]);
		if(i==x[j])rx[j]=g[i];
		else ry[j]=g[i];
	}
	g[z]=su(g[z],f[z]);
	//printf("a%d %d %d\n",z,f[z],g[z]);
}

void dfs2(int z,int fa,ll f1,ll g1){
	//printf("NMSL%d %d %d %d\n",z,fa,f1,g1);
	int i,j,k,n=v[z].size(),m;
	vector<ll> vl(n+5),vr(n+5);
	m=0;
	ll f2,g2;
	for(auto [i,j]:v[z]){
		if(i==fa)continue;
		g1=su(g1,g[i]);
		m++;
		vl[m]=vr[m]=f[i]+1;
	}
	vl[0]=vr[m+1]=1;
	for(i=1;i<=m;i++){
		vr[m-i]=vr[m-i]*vr[m-i+1]%M;
		vl[i]=vl[i]*vl[i-1]%M;
		//printf("nmsl%d %d %d %d\n",z,i,vl[i],vr[i]);
	}
	k=0;
	for(auto [i,j]:v[z]){
		if(i==fa)continue;
		k++;
		f2=(f1+1)*vl[k-1]%M*vr[k+1]%M;
		g2=su(g1,M-g[i]);
		g2=su(g2,f2);
		if(i==x[j])ry[j]=g2;
		else rx[j]=g2;
		dfs2(i,z,f2,g2);
	}
}

int main(){
	ios::sync_with_stdio(0);
	cin>>n;
	for(i=1;i<n;i++){
		cin>>x[i]>>y[i];
		v[x[i]].push_back({y[i],i});
		v[y[i]].push_back({x[i],i});
	}
	dfs(1,0);
	dfs2(1,0,0,0);
	for(i=1;i<n;i++){
		cout<<rx[i]<<' '<<ry[i]<<'\n';
	}
}