Newcoder 83 D.队列重排（dp+组合数学）

Description

有$n(n\le 500000)$ 个人排成一列，把他们解散后重排，使得"重排后前方" 跟"原排列前方" 一样的人不超过$k(k<n)$ 个，问有几种方法数，答案请$mod(10^9+7)$ 输出。

举例来说，有五个人编号为$1$∼$5$ 间的整数，最初的排列由前至后依序为$1,2,3,4,5$，重排列后顺序由前至后变为$1,3,4,2,5$，其中只有编号为$4$ 的人，“原排列前方” 跟"重排后前方" 都是编号为$3$ 的人，故"重排后前方" 跟"原排列前方" 一样的人只有$1$人。

原排列的第$1$个人和重排后的第$1$个人一定不会是「“重排后前方” 跟 “原排列前方” 一样的人」。

Input

输入只有一行，有两个正整数$n,k$，满足$1\le k<n\le 500000$。

Output

请输出一行包含一个小于$10^9+7 $的非负整数，表示总共的方法数 $mod(10^9+7)$。

Sample Input

3 1

Sample Output

5

Solution

令$a_n$为$n$个人重排后前方的人均不是原排列前方人的方案数，假设起初排位为$1,...,n$，考虑$1$的作用

$1.$若$1$不再任意一对$i,i+1$之间，那么需要$2$~$n$这$n-1$个人重排后前方的人均不是原排列前方的人，方案数为$a_{n-1}$，对于这其中的每一种方案，只要$1$不在$2$前方即可，故$1$有$n-1$个位置可以放

$2.$若$1$在某对$i,i+1$之间，选择一个$i$方案数为$n-2$，之后把$i,1,i+1$看作一个元素$i+1$，问题转化为$n-2$个人重排后每个人前方的人均不是原排列前方的人，方案数$a_{n-2}$

综上有$a_n=(n-1)\cdot a_{n-1}+(n-2)\cdot a_{n-2}$，线性预处理$a_n$即可

之后考虑原问题，假设重排后有$x$个人前方的人和原排列前方的人相同，从$n-1$个有前方人的人中选出这$x$个人方案数$C_{n-1}^x$，选出后，把这$x$对看作$x$个元素，问题转化为$n-x$个人重排后每个人前方的人均不是原排列前方人的方案数，即$a_{n-x}$，故答案为$\sum _{x=0}^k{C}_{n-1}^x\cdot a_{n-x}$

Code

#include<cstdio>
using namespace std;
typedef long long ll;
const int maxn=500005;
#define mod 1000000007
int add(int x,int y)
{
	x+=y;
	if(x>=mod)x-=mod;
	return x;
}
int mul(int x,int y)
{
	ll z=1ll*x*y;
	return z-z/mod*mod;
}
int n,k,a[maxn],inv[maxn],fact[maxn];
void init(int n=5e5)
{
	inv[1]=1;
	for(int i=2;i<=n;i++)inv[i]=mul(mod-mod/i,inv[mod%i]);
	inv[0]=1;
	for(int i=1;i<=n;i++)inv[i]=mul(inv[i-1],inv[i]);
	fact[0]=1;
	for(int i=1;i<=n;i++)fact[i]=mul(i,fact[i-1]);
}
int C(int n,int m)
{
	return mul(fact[n],mul(inv[m],inv[n-m]));
}
int main()
{
	init();
	scanf("%d%d",&n,&k);
	a[1]=1,a[2]=1;
	for(int i=3;i<=n;i++)a[i]=add(mul(i-1,a[i-1]),mul(i-2,a[i-2]));
	int ans=0;
	for(int i=0;i<=k;i++)ans=add(ans,mul(C(n-1,i),a[n-i]));
	printf("%d\n",ans);
	return 0;
}