【CodeChef-TREEWALK】Walk on Tree（BM算法）（特征多项式）

传送门

---

题解：

首先由于求的是恰好$K$次，所以直接矩阵快速幂的做法大家应该都会

我们发现实际上就是求矩阵中一些位置的和。

然后这玩意居然有递推式，而且对于所有位置，这个递推式是相同的！

而且这个递推式其实就是矩阵的特征多项式。

为什么，其实很简单，考虑我们直接做线性递推用的其实就是递推矩阵的特征多项式，而由凯莱-哈密顿定理，一个矩阵的特征多项式是一个零化多项式。

设$S(A)$表示对矩阵$A$的一个位置集合求和。

现在我们要求的是$S(A^k)$，由于其特征多项式为零化多项式，则原矩阵减去任意次特征多项式，求和结果不变，对于任意位置都是如此，于是有$S(A^k)=S(A^k\%Z)$。

我建议看不懂的先去看一下矩阵多项式的定义。

求一般的矩阵的特征多项式是$O(n^4)$，因为需要求$O(n)$个矩阵乘法。

然而这个稀疏矩阵只有$O(n)$个位置上有值，直接暴力算出即可。

---

代码：

#include
#define ll long long
#define re register
#define gc get_char
#define cs const

namespace IO{
	inline char get_char(){
		static cs int Rlen=1<<22|1;
		static char buf[Rlen],*p1,*p2;
		return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++;
	}
	
	template<typename T>
	inline T get(){
		char c;
		while(!isdigit(c=gc()));T num=c^48;
		while(isdigit(c=gc()))num=(num+(num<<2)<<1)+(c^48);
		return num;
	}
	inline int getint(){return get<int>();}
}
using namespace IO;

using std::cerr;
using std::cout;

cs int mod=998244353;
inline int add(int a,int b){return (a+=b)>=mod?a-mod:a;}
inline int dec(int a,int b){return (a-=b)<0?a+mod:a;}
inline int mul(int a,int b){static ll r;r=(ll)a*b;return r>=mod?r%mod:r;}
inline int power(int a,int b,int res=1){
	for(;b;b>>=1,a=mul(a,a))(b&1)&&(res=mul(res,a));
	return res;
}
inline void Inc(int &a,int b){(a+=b)>=mod&&(a-=mod);}
inline void Dec(int &a,int b){(a-=b)<0&&(a+=mod);}
inline void Mul(int &a,int b){a=mul(a,b);} 

typedef std::vector<int> Poly;

cs int bit=20,SIZE=1<<bit|1;

int r[SIZE],*w[bit+1];
inline void init_NTT(){
	for(int re i=1;i<=bit;++i)w[i]=new int[1<<(i-1)];
	int wn=power(3,mod-1>>bit);
	w[bit][0]=1;for(int re i=1;i<(1<<bit-1);++i)w[bit][i]=mul(w[bit][i-1],wn);
	for(int re i=bit-1;i;--i)
	for(int re j=0;j<(1<<i-1);++j)w[i][j]=w[i+1][j<<1];
}
inline void NTT(Poly &A,int len,int typ){
	for(int re i=0;i<len;++i)if(i<r[i])std::swap(A[i],A[r[i]]);
	for(int re i=1,d=1;i<len;i<<=1,++d)
	for(int re j=0;j<len;j+=i<<1)
	for(int re k=0;k<i;++k){
		int &t1=A[j+k],&t2=A[i+j+k],t=mul(t2,w[d][k]);
		t2=dec(t1,t),Inc(t1,t);
	}
	if(typ==-1){
		std::reverse(A.begin()+1,A.begin()+len);
		for(int re i=0,inv=power(len,mod-2);i<len;++i)Mul(A[i],inv);
	}
}
inline void init_rev(int l){
	for(int re i=0;i<l;++i)r[i]=r[i>>1]>>1|((i&1)?l>>1:0);
}

inline Poly operator+(cs Poly &a,cs Poly &b){
	Poly c=a;if(b.size()>a.size())c.resize(b.size());
	for(int re i=0,li=b.size();i<li;++i)Inc(c[i],b[i]);
	return c;
}

inline Poly operator-(cs Poly &a,cs Poly &b){
	Poly c=a;if(b.size()>a.size())c.resize(b.size());
	for(int re i=0,li=b.size();i<li;++i)Dec(c[i],b[i]);
	return c;
}

inline Poly operator*(cs Poly &c,cs Poly &d){
	if(!c.size()||!d.size())return Poly(0,0);
	int deg=c.size()+d.size()-1,l=1;Poly a=c,b=d;
	while(l<deg)l<<=1;init_rev(l);
	a.resize(l),NTT(a,l,1);
	b.resize(l),NTT(b,l,1);
	for(int re i=0;i<l;++i)Mul(a[i],b[i]);
	NTT(a,l,-1),a.resize(deg);
	return a;
}

inline Poly Inv(cs Poly &a,int lim){
	int n=a.size();Poly c,b(1,power(a[0],mod-2));
	for(int re l=4;(l>>2)<lim;l<<=1){
		init_rev(l);
		c.resize(l>>1);for(int re i=0;i<(l>>1);++i)c[i]=i<n?a[i]:0;
		c.resize(l),NTT(c,l,1);
		b.resize(l),NTT(b,l,1);
		for(int re i=0;i<l;++i)Mul(b[i],dec(2,mul(b[i],c[i])));
		NTT(b,l,-1),b.resize(l>>1);
	}b.resize(lim);
	return b;
}

inline Poly operator/(cs Poly &c,cs Poly &d){
	if(c.size()<d.size())return Poly(0,0);
	Poly a=c,b=d;int deg=c.size()-d.size()+1,l=1;
	std::reverse(a.begin(),a.end());
	std::reverse(b.begin(),b.end());
	while(l<deg)l<<=1;
	b=Inv(b,l),b.resize(deg);
	a=a*b;a.resize(deg);
	std::reverse(a.begin(),a.end());
	return a;
}

inline Poly operator%(cs Poly &a,cs Poly &b){
	if(a.size()<b.size())return a;
	Poly c=a-(a/b)*b;
	c.resize(b.size()-1);
	return c;
}

inline Poly Ksm(Poly a,int b,cs Poly &mod){
	Poly res(1,1);
	for(;b;b>>=1,a=a*a%mod)
	if(b&1)res=res*a%mod;
	return res;
}

cs int N=3e3+50,M=6.1e3+50;

namespace BM{
	int L,cnt,a[M],fail[M],delta[M];
	Poly R[M];
	
	inline Poly solve(){
		for(int re i=1;i<=L;++i){
			int d=a[i];
			for(int re j=1,lj=R[cnt].size();j<lj;++j)Dec(d,mul(R[cnt][j],a[i-j]));
			if(!d)continue;
			fail[cnt]=i,delta[cnt]=d;
			if(!cnt){R[++cnt].resize(i+1);continue;}
			else {
				int coef=mul(d,power(delta[cnt-1],mod-2));
				R[cnt+1].resize(i-fail[cnt-1]);R[cnt+1].push_back(coef);
				for(int re j=1,lj=R[cnt-1].size();j<lj;++j)R[cnt+1].push_back(mul(mod-coef,R[cnt-1][j]));
				R[cnt+1]=R[cnt+1]+R[cnt];++cnt;
			}
		}
		return R[cnt];
	}
}

int n,K,rt;
std::vector<int> G[N];
int dp[N][M];

Poly f,g;

signed main(){
#ifdef zxyoi
	freopen("treewalk.in","r",stdin);
#endif
init_NTT();
	n=getint();BM::L=n*2+15;
	for(int re i=1;i<n;++i){
		int u=getint(),v=getint();
		G[u].push_back(v);
		G[v].push_back(u);
	}
	rt=getint(),K=getint();
	dp[rt][0]=1;
	for(int re i=0;i<BM::L;++i){
		for(int re u=1;u<=n;++u)if(dp[u][i])
		for(int re v:G[u])Inc(dp[v][i+1],dp[u][i]);
	}
	for(int re i=1;i<=BM::L;++i){BM::a[i]=dp[rt][i-1];}
	f=BM::solve();std::reverse(f.begin(),f.end());
	for(int re i=0,li=f.size();i<li;++i)f[i]=dec(0,f[i]);f.back()=1;
	g.resize(2),g[1]=1;g=Ksm(g,K,f);
	for(int re i=1;i<=n;++i){
		int ans=0;
		for(int re j=0;j<g.size();++j)Inc(ans,mul(dp[i][j],g[j]));
		cout<<ans<<" ";
	}
	return 0;
}