换根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';
}
}