队列重排

题目:
https://ac.nowcoder.com/acm/problem/15614

有$n(n\le 500000)$个人排成一列，把他们解散后重排，使得"重排后前方" 跟"原排列前方" 一样的人不超过$k(k<n)$个，问有几种方法数，答案请$mod(10^9+7)$输出。举例来说，有五个人编号为$1\sim 5$间的整数，最初的排列由前至后依序为$1,2,3,4,5$，重排列后顺序由前至后变为$1,3,4,2,5$，其中只有编号为$4$的人，“原排列前方” 跟"重排后前方" 都是编号为$3$的人，故"重排后前方" 跟"原排列前方" 一样的人只有$1$人。原排列的第$1$个人一定没有。

思路：

设$f(i)$表示"重排后前方" 跟"原排列前方" 一样的人有$i$个的方案数。刚开始有$n$个连通块，就是$1\sim n$之间没有限制，如果编号为$j(2\le j)$的人重排之后前面还是$j-1$，则$j$和$j-1$一定要绑定在一起，则连通块个数减$1$。所以$f(i)$的时候有$n-i$个连通块，所以有公式，$n=$相邻的个数$+$连通块的个数。那么现在就还得重排这$n-i$个连通块，保证连通块没有"重排后前方" 跟"原排列前方" 一样(因为连通块里面已经保证了有$i$个相邻，那么连通块之间就不能相邻)。

设$h(i)$表示$i$个连通块不存在"重排后前方" 跟"原排列前方" 一样的方案数。然后考虑递推，假设已经放好了前$i-1$个，这$i-1$个满足不存在"重排后前方" 跟"原排列前方" 一样。那么最后一个连通块可以有$i-1$个位置放。但是还有另外一种情况就是$i-1$个当中存在一个"重排后前方" 跟"原排列前方" 一样(有$i-2$种方式)，然后最后一个插入进去破坏掉。这样相当于前$i-1$个只有$i-2$个连通块。所以有递推式
$$h(i)=(i-1)h(i-1)+(i-2)h(i-2)$$
则
$$\begin{array}{rl}f(i)& =\left(\genfrac{}{}{0ex}{}{n}{i}\right)h(n-i)\\ ans& =\sum _{i=0}^{k}f(i)\end{array}$$

#include<cstdio>
const int N=5e5+88;
const int P=1e9+7;
int ksm(int x,int k){
	int ans=1;
	for(;k;k>>=1,x=1LL*x*x%P) if(k&1) ans=1LL*ans*x%P;
	return ans;
}
int ans,n,k,C[N],a[N];
int main(){ 
    scanf("%d%d",&n,&k);
    a[1]=a[2]=1;
    for(int i=3;i<=n;++i) a[i]=(1LL*(i-1)*a[i-1]%P+1LL*(i-2)*a[i-2]%P)%P;
    C[0]=1;
    for(int i=1;i<n;++i) C[i]=1LL*C[i-1]*(n-i)%P*ksm(i,P-2)%P;
    for(int i=0;i<=k;++i) ans=(ans+1LL*C[i]*a[n-i]%P)%P;
    printf("%d\n",ans);
}