JZOJ6830. 【2020.10.25提高组模拟】排列

Description

• $n\le 1000$

Solution

• 梅开二度，几个月前才做过相同套路的题目，结果由于对于容斥的理解不够透彻，最后没有想清楚乱推出来的东西的正确性。
• 首先可以发现一个结论，对于一个排列，找到最长长度为$len$的上升区间，它需要操作的次数是$n-len$。
• 那么容易转化为对于每一个$k$，求$maxlen\le k$的排列的个数。
• 直接模拟DP可以做到$O(n^4)$或$O(n^5)$
• 一个很$naive$的做法是往当前的排列里面接一些连续上升的段，再直接乘上分配的组合数。
• 但是显然无法保证头尾会不会连接成更大的段，所以我们对于每一个长度为$len$的段都要给它一个容斥系数$f(len)$使得，在分配完标号以后如果组合成长度大于$k$的段，贡献就是$0$。
• 写成生成函数的形式，$F(x)$为$f$的生成函数，即

$$\sum _{i=0}^{inf}{F}^i(x)=\sum _{i=1}^k{x}^i$$

$$1+\sum _{i=0}^{inf}{F}^i(x)=\sum _{i=0}^k{x}^i$$

$$\frac{1}{1-F(x)}=\frac{x^{k+1}-1}{x-1}$$

$$F(x)=1-\frac{x-1}{x^{k+1}-1}$$

$$F(x)=1+(x-1)\frac{1}{1-x^{k+1}}$$

$$F(x)=\sum _{i\ge 0}{x}^{(k+1)i+1}-\sum _{i\ge 1}{x}^{(k+1)i}$$

• 由于转移函数$F(x)$只有$\frac{n}{k}$项，所以暴力卷积，复杂度是$O(n\ast \frac{n}{k})$。
• 总复杂度就是$O(n^{2}logn)$的。

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define maxn 1005
#define ll long long 
using namespace std;

int n,mo,i,j,k;
ll fct[maxn],invf[maxn],ans[maxn],f[maxn];

ll ksm(ll x,ll y){
	ll s=1;
	for(;y;y/=2,x=x*x%mo) if (y&1)
		s=s*x%mo;
	return s;
}

int main(){
//	freopen("ceshi.in","r",stdin);
	freopen("permutation.in","r",stdin);
	freopen("permutation.out","w",stdout);
	scanf("%d%d",&n,&mo);
	fct[0]=1;for(i=1;i<maxn;i++) fct[i]=fct[i-1]*i%mo;
	invf[maxn-1]=ksm(fct[maxn-1],mo-2);
	for(i=maxn-2;i>=0;i--) invf[i]=invf[i+1]*(i+1)%mo;
	for(k=1;k<n;k++){
		memset(f,0,sizeof(f));
		f[0]=1;
		for(i=0;i<n;i++) {
			for(j=0;i+(k+1)*j+1<=n;j++)	
				(f[i+(k+1)*j+1]+=f[i]*invf[(k+1)*j+1])%=mo;
			for(j=1;i+(k+1)*j<=n;j++)
				(f[i+(k+1)*j]-=f[i]*invf[(k+1)*j])%=mo;
		}
		ans[k]=f[n]*fct[n]%mo;
	}
	ans[n]=fct[n]; ll s=0;
	for(i=0;i<n;i++){
		(s+=ans[n-i]-ans[n-i-1])%=mo;
		printf("%lld\n",(s+mo)%mo);
	}
}