AtCoder Grand Contest 005 D - ~K Perm Counting 容斥原理+dp

题意

给出n和k，问有多少个长度为n的排列p使得对于任意的i，有$|p_i-i|\not=k$。
n<=2000

分析

正着不好做，考虑容斥。
那么我们要求的就是至少选x个位置不满足条件的方案。
注意到每个数和他相减绝对值为k的数只有不超过两个，也就是说，如果我们在x和x+k之间连一条边，那么n个点会形成很多条链，每个点只能和与它有边相连的点配对。
把这若干条链排成一排。
设f[i,j]表示前i个点，已经选了j对不满足条件的数出来的方案。
注意到i可能和它的上一个点和下一个点配对，同时i的下一个点可能会与i配对，又因为每个数只能被配对一次，所以我们需要记录一下当前点和下一个点是否已经被配对。
那就是f[i,j,0/1,0/1]表示前i个点，选了j对不满足的数出来，当前数和下一个数是否被配对的方案。
转移显然。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=2005;
const int MOD=924844033;

int n,k,jc[N],a[N];
LL f[N][N][2][2];
bool vis[N];

int main()
{
    scanf("%d%d",&n,&k);
    jc[0]=1;
    for (int i=1;i<=n;i++) jc[i]=(LL)jc[i-1]*i%MOD;
    int tot=0;
    for (int i=1;i<=n;i++)
        if (!vis[i])
        {
            a[++tot]=i;
            for (int j=i+k;j<=n;j+=k) vis[j]=1,a[++tot]=j;
        }
    f[0][0][0][0]=1;a[0]=-k-1;a[n+1]=k-1;
    for (int i=1;i<=n;i++)
    {
        f[i][0][0][0]=1;
        for (int j=1;j<=i;j++)
        {
            f[i][j][0][0]=(f[i-1][j][1][0]+f[i-1][j][0][0]+(a[i]-a[i-1]==k)*f[i-1][j-1][0][0])%MOD;
            f[i][j][0][1]=(a[i+1]-a[i]==k)*(f[i-1][j-1][1][0]+f[i-1][j-1][0][0])%MOD;
            f[i][j][1][0]=(f[i-1][j][0][1]+f[i-1][j][1][1]+(a[i]-a[i-1]==k)*f[i-1][j-1][0][1])%MOD;
            f[i][j][1][1]=(a[i+1]-a[i]==k)*(f[i-1][j-1][0][1]+f[i-1][j-1][1][1])%MOD;
        }
    }
    int ans=0;
    for (int i=0;i<=n;i++)
    {
        int s=(f[n][i][0][0]+f[n][i][0][1]+f[n][i][1][0]+f[n][i][1][1])%MOD;
        if (i&1) (ans+=MOD-(LL)s*jc[n-i]%MOD)%=MOD;
        else (ans+=(LL)s*jc[n-i]%MOD)%=MOD;
    }
    printf("%d",ans);
    return 0;
}