【LOJ2463】「2018 集训队互测 Day 1」完美的旅行（BM算法）（多项式求逆）（FMT）

传送门

---

题解：

显然我们的想法是对所有集合算出它的超集的答案然后FMT回去。

发现转移是一个矩阵乘法的形式，显然有线性递推式。

直接用BM求出线性递推式可以算出走单次$k$步，愉悦值为$i$的方案数。

然后FMT得到单次走$k$步，愉悦值为$i$的超集的方案数，记为$g(i,k)$，设其OGF为$G_i(x)$。

现在需要求总共走$k$步，愉悦值为$i$的超集的方案数，记为$f(i,k)$，设其OGF为$F_i(x)$。

显然有$$F_i(x)=1+G_i(x)+G_i^2(x)+\cdots =\frac{1}{1-G_i(x)}$$

多项式求逆后再FMT回去即可。

---

代码：

#include<bits/stdc++.h>
#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 gi(){return get<int>();}
}
using namespace IO;

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

cs int mod=998244353;
inline int add(int a,int b){a+=b-mod;return a+(a>>31&mod);}
inline int dec(int a,int b){a-=b;return a+(a>>31&mod);}
inline int mul(int a,int b){ll 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+=(a>>31&mod);}
inline void Dec(int &a,int b){a-=b;a+=(a>>31&mod);}
inline void Mul(int &a,int b){a=mul(a,b);}

using Poly=std::vector<int>;

cs int bit=16,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(int *A,int len,int typ){
	for(int re i=1;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[j+k+i],t=mul(t2,w[d][k]);
		t2=dec(t1,t),Inc(t1,t);
	}
	if(typ==-1){
		std::reverse(A+1,A+len);
		for(int re i=0,inv=power(len,mod-2);i<len;++i)Mul(A[i],inv);
	}
}
inline void NTT(Poly &A,int len,int typ){NTT(&A[0],len,typ);}
inline void init_rev(int l){
	for(int re i=1;i<l;++i)r[i]=r[i>>1]>>1|((i&1)?l>>1:0);
}

inline Poly Inv(cs Poly &a,int lim=-1){
	int n=a.size();Poly c,b(1,power(a[0],mod-2));if(lim==-1)lim=n;
	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 &a,cs Poly &b){
	Poly c=a;if(a.size()<b.size())c.resize(b.size());
	for(int re i=0;i<b.size();++i)Inc(c[i],b[i]);
	return c;
}

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

namespace BM{
	cs int N=1000;
	int cnt,fail[N],delta[N];
	Poly R[N];
	inline Poly solve(int L,int *a){
		R[cnt=0].clear();
		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(a[i-j],R[cnt][j]));
			if(!d)continue;
			fail[cnt]=i,delta[cnt]=d;
			if(!cnt){++cnt;R[cnt].clear();R[cnt].resize(i+1);}
			else {
				R[cnt+1].clear();
				int coef=mul(delta[cnt],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];
	}
}

cs int N=70;

int n,m;
int A[N][N];
int way[N*2+15][N][N];
int a[N*2+15];
Poly p[N],ans[N],h;

signed main(){
#ifdef zxyoi
	freopen("travel.in","r",stdin);
#endif
	init_NTT();
	n=gi(),m=gi();int L=2*n+5,lim=std::max(L,m)+1;
	for(int re i=0;i<n;++i){
		way[0][i][i]=1;
		for(int re j=0;j<n;++j)A[i][j]=gi();
		p[i].resize(lim+1);
	}
	for(int re t=1;t<=L;++t){
		for(int re i=0;i<n;++i)
		for(int re j=0;j<n;++j)if(int v=way[t-1][i][j])
		for(int re k=0;k<n;++k)Inc(way[t][i][k],mul(v,A[j][k]));
		for(int re i=0;i<n;++i)
		for(int re j=0;j<n;++j)Inc(p[i&j][t],way[t][i][j]);
	}
	for(int re t=1;t<=L;++t)
	for(int re i=0;i<n;++i)Mul(a[t],12345),Inc(a[t],p[i][t]);
	h=BM::solve(L,a);
	for(int re i=0;i<n;++i)
	for(int re t=L+1;t<=m;++t)
	for(int re j=1;j<h.size();++j)Inc(p[i][t],mul(p[i][t-j],h[j]));
	for(int re i=0;i<n;++i){
		for(int re t=1;t<=m;++t)
		for(int re j=i+1;j<n;++j)
		if((i|j)==j)Inc(p[i][t],p[j][t]);
		p[i][0]=1;
		for(int re j=1;j<=m;++j)p[i][j]=dec(0,p[i][j]);
		ans[i]=Inv(p[i]);
	}
	int res=0;
	for(int t=1;t<=m;++t){
		for(int re i=n;i--;){
			for(int re j=i+1;j<n;++j)
			if((i|j)==j)Dec(ans[i][t],ans[j][t]);
			res^=ans[i][t];
		}
	}
	cout<<res<<"\n";
	return 0;
}