51nod1514 美妙的序列 分治NTT

Description

---

某个1~n的排列如果满足：
在1~n-1 这些位置后面将序列断开，使得总可以从右边找到一个数，并且该数不大于左边的所有数，则称该序列为“美妙的”。
给出n，求长度为n的“美妙的序列”的数量Mod 998244353
$T,n\le 10^5$

Solution

---

考虑设f[n]表示长度为n的美妙序列，我们用总的方案减去不合法的方案就是$f[n]=n!-\sum f[i]\cdot (n-i)!$
于是这就是一个非常经典的分治NTT，当然如果你移一下项写多项式求逆也是没问题的

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (register int i=st;i<=ed;++i)

typedef long long LL;
const int MOD=998244353;
const int N=5000005;

LL fac[N],A[N],B[N],f[N];

int rv[N];

int read() {
	int x=0,v=1; char ch=getchar();
	for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
	for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
	return x*v;
}

void write(LL x) {
	if (x>=10) write(x/10);
	putchar(x%10+'0');
}

void swap(LL &x,LL &y) {
	LL tmp=x; x=y; y=tmp;
}

LL ksm(LL x,LL dep) {
	LL res=1;
	for (;dep;dep>>=1) {
		(dep&1)?(res=res*x%MOD):0;
		x=x*x%MOD;
	}
	return res;
}

void NTT(LL *a,int n,int f) {
	for (int i=0;i<n;++i) if (i<rv[i]) swap(a[i],a[rv[i]]);
	for (int i=1;i<n;i<<=1) {
		LL wn=ksm(3,(f==1)?((MOD-1)/i/2):(MOD-1-(MOD-1)/i/2));
		for (int j=0;j<n;j+=(i<<1)) {
			LL w=1;
			for (int k=0;k<i;++k) {
				LL u=a[j+k],v=a[j+k+i]*w%MOD;
				a[j+k]=u+v; (a[j+k]>=MOD)?(a[j+k]-=MOD):0;
				a[j+k+i]=u+MOD-v; (a[j+k+i]>=MOD)?(a[j+k+i]-=MOD):0;
				w=w*wn%MOD;
			}
		}
	}
	if (f==-1) {
		LL ny=ksm(n,MOD-2);
		for (int i=0;i<n;++i) a[i]=a[i]*ny%MOD;
	}
}

void solve(int l,int r) {
	if (l==r) {
		if (l>1) f[l]=fac[l]-f[l],(f[l]<0)?(f[l]+=MOD):0;
		return ;
	}
	int mid=(l+r)>>1;
	solve(l,mid);
	int len=1,lg=0; for (;len<=(r-l+1)*2;) len<<=1,++lg;
	for (int i=0;i<len;++i) {
		rv[i]=(rv[i>>1]>>1)|((i&1)<<lg-1);
		A[i]=0,B[i]=fac[i];
	}
	rep(i,l,mid) A[i-l]=f[i];
	NTT(A,len,1); NTT(B,len,1);
	for (int i=0;i<len;++i) A[i]=A[i]*B[i]%MOD;
	NTT(A,len,-1);
	rep(i,mid+1,r) f[i]=f[i]+A[i-l],(f[i]>=MOD)?(f[i]-=MOD):0;
	solve(mid+1,r);
}

int main(void) {
	freopen("data.in","r",stdin);
	freopen("std.out","w",stdout);
	fac[0]=fac[1]=1;
	rep(i,2,1e5) fac[i]=fac[i-1]*i%MOD;
	f[0]=0; f[1]=1;
	solve(0,1e5);
	for (int T=read(),x;T--;) {
		x=read();
		write(f[x]); putchar('\n');
	}
	return 0;
}