CF285E Positions in Permutations(dp+容斥)

题意，给定n,k,求有多少排列是的 | p[i]-i |=1 的数量为k。

Solution

直接dp会有很大的后效性。

所以我们考虑固定k个数字使得它们是合法的，所以我们设dp[i][j][0/1][0/1]表示前i个数，填了j个数，当前位置有没有被选，下一位有没有被选，这样做的话，转移会比较简单。

那么除去这j个数，剩下的数随便填，乘上全排列就好了。

但这样会多算。

然后这种问题有一个容斥模型，直接套上就好了。

#include<iostream>
#include<cstdio>
#define N 1002
using namespace std;
typedef long long ll;
int n,k;
ll dp[N][N][][],jie[N],ni[N],g[N],ans;
const int mod=1e9+;
ll calc(int n,int m){
    return jie[n]*ni[m]%mod*ni[n-m]%mod;
}
ll power(ll x,int y){
    ll ans=;
    while(y){
        if(y&)(ans*=x)%=mod;
        (x*=x)%=mod;
        y>>=;
    }
    return ans;
}
int main(){
    scanf("%d%d",&n,&k);jie[]=;
    for(int i=;i<=n;++i)jie[i]=(jie[i-]*i)%mod;ni[n]=power(jie[n],mod-);
    for(int i=n-;i>=;--i)ni[i]=ni[i+]*(i+)%mod;
    dp[][][][]=;
    for(int i=;i<=n;++i){
        for(int j=;j<=n;++j){
            dp[i][j][][]=(dp[i-][j][][]+dp[i-][j][][])%mod;
            dp[i][j][][]=(dp[i-][j][][]+dp[i-][j][][])%mod;
            if(j){
            (dp[i][j][][]+=dp[i-][j-][][])%=mod;
            dp[i][j][][]+=(dp[i-][j-][][]+dp[i-][j-][][])%mod;
            dp[i][j][][]%=mod;
            (dp[i][j][][]+=dp[i-][j-][][])%=mod;
            dp[i][j][][]+=(dp[i-][j-][][]+dp[i-][j-][][])%mod;
            dp[i][j][][]%=mod;
            }
        }
    }
    for(int i=k;i<=n;++i)
      g[i]=(dp[n][i][][]+dp[n][i][][])%mod*jie[n-i]%mod;
    for(int i=k;i<=n;++i)(ans+=(((i-k)&)?-:)*calc(i,k)*g[i]%mod+mod)%=mod;
    ans=(ans+mod)%mod;
    cout<<ans;
    return ;
}