Description
定义一个排列的权为满足
|pi−i|=1
的
i
的个数,问长度为
Input
两个整数
Output
输出满足条件的排列数,结果模 109+7
Sample Input
3 2
Sample Output
4
Solution
第
i
个位置想合法必然要放
dp[i][j+1][y][0]+=dp[i−1][j][0][y]
dp[i][j+1][y][1]+=dp[i−1][j][x][y],i<n
dp[i][j][y][0]=dp[i−1][j][x][y]
令
f[i]=(n−i)!⋅∑x=01∑y=01dp[n][i][x][y]
,则
f[i]
为权至少为
i
的排列个数,那么对于
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=1005;
#define mod 1000000007
int C[maxn][maxn],fact[maxn];
void add(int &x,int y)
{
x=x+y>=mod?x+y-mod:x+y;
}
void sub(int &x,int y)
{
x=x-y<0?x-y+mod:x-y;
}
void init(int n=1000)
{
C[0][0]=1;
for(int i=1;i<=n;i++)
{
C[i][0]=C[i][i]=1;
for(int j=1;j<i;j++)add(C[i][j],C[i-1][j-1]+C[i-1][j]);
}
fact[0]=1;
for(int i=1;i<=n;i++)fact[i]=(ll)i*fact[i-1]%mod;
}
int n,k,dp[maxn][maxn][2][2],f[maxn];
int main()
{
init();
scanf("%d%d",&n,&k);
dp[0][0][1][0]=1;
for(int i=1;i<=n;i++)
for(int j=0;j<i;j++)
for(int x=0;x<=1;x++)
for(int y=0;y<=1;y++)
{
if(!x)add(dp[i][j+1][y][0],dp[i-1][j][x][y]);
if(i<n)add(dp[i][j+1][y][1],dp[i-1][j][x][y]);
add(dp[i][j][y][0],dp[i-1][j][x][y]);
}
for(int i=0;i<=n;i++)
for(int x=0;x<=1;x++)
for(int y=0;y<=1;y++)
add(f[i],dp[n][i][x][y]);
for(int i=0;i<=n;i++)f[i]=(ll)f[i]*fact[n-i]%mod;
for(int i=n-1;i>=k;i--)
for(int j=i+1;j<=n;j++)
sub(f[i],(ll)f[j]*C[j][i]%mod);
printf("%d\n",f[k]);
return 0;
}