[Codeforces285E]Positions in Permutations（dp+容斥原理+组合数学）

题目描述

传送门

题解

感觉这题难在dp啊
f(i,j,0/1,0/1/2)表示前i个中，选出j个满足要求，第i个位置有/没有，第i个不选/放在i-1/放在i+1的方案数
dp的具体方法看代码吧，感觉状态没法减了
求出了这个之后相当于就求出了前n个选i个满足要求的方案数，剩下的就容斥一下就好啦（乘全排）
容斥系数是$C_i^k$（貌似选k个的都是这个）

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define Mod 1000000007
#define LL long long
#define N 1005

int n,k;
LL mul[N],c[N][N],f[N][N][2][3],chose[N];
LL ans;

int main()
{
    scanf("%d%d",&n,&k);
    mul[0]=1;for (int i=1;i<=n;++i) mul[i]=mul[i-1]*(LL)i%Mod;
    for (int i=0;i<=1000;++i) c[i][0]=1;
    for (int i=1;i<=1000;++i)
        for (int j=1;j<=1000;++j)
            c[i][j]=(c[i-1][j]+c[i-1][j-1])%Mod;
    for (int i=0;i<=n;++i) f[i][0][1][0]=1;
    for (int i=1;i<=n;++i)
        for (int j=0;j<=i;++j)
        {
            // 第i个为空，第i个数不选：从前i-1里选j个，第i-1个数不能放在右边 
            f[i][j][0][0]=f[i-1][j][0][0]+f[i-1][j][0][1]+f[i-1][j][1][0]+f[i-1][j][1][1];
            f[i][j][0][0]%=Mod;
            // 第i个为空，第i个数放在左边：从前i-1里选j-1个，第i-1个数不能放在右边，第i-1个格子必须为空 
            if (j) f[i][j][0][1]=f[i-1][j-1][0][0]+f[i-1][j-1][0][1];
            f[i][j][0][1]%=Mod;
            // 第i个为空，第i个数放在右边：从前i-1里选j-1个，第i-1个数不能放在右边
            if (j) f[i][j][0][2]=f[i-1][j-1][0][0]+f[i-1][j-1][0][1]+f[i-1][j-1][1][0]+f[i-1][j-1][1][1];
            f[i][j][0][2]%=Mod;

            // 第i个不为空，第i个数不选：从前i-1里选j个，第i-1个数放在右边
            f[i][j][1][0]=f[i-1][j][0][2]+f[i-1][j][1][2];
            f[i][j][1][0]%=Mod;
            // 第i个不为空，第i个数放在左边：从前i-1里选j-1个，第i-1个数放在右边，第i-1个格子必须为空
            if (j) f[i][j][1][1]=f[i-1][j-1][0][2];
            f[i][j][1][1]%=Mod;
            // 第i个不为空，第i个数放在右边：从前i-1里选j-1个，第i-1个数放在右边
            if (j) f[i][j][1][2]=f[i-1][j-1][0][2]+f[i-1][j-1][1][2];
            f[i][j][1][2]%=Mod;
        }
    for (int i=0;i<=n;++i)
    {
        chose[i]=f[n][i][0][0]+f[n][i][0][1]+f[n][i][1][0]+f[n][i][1][1];
        chose[i]%=Mod;
    }
    for (int i=k,opt=1;i<=n;++i,opt=-opt)
    {
        ans+=chose[i]*mul[n-i]%Mod*c[i][k]%Mod*opt;
        ans=(ans%Mod+Mod)%Mod;
    }
    printf("%I64d\n",ans);
}