【agc005d】~K Perm Counting

题目大意

求有多少中1~n的排列，使得\(abs(第i个位置的值-i)!=k\)

解题思路

考虑容斥，\(ans=\sum_{i=0}^{n}(-1)^ig[i](n-i)!(g[i]表示至少有i个位置是不合法的方案数)\)

考虑如何求g[i]

将每个位置和每个值都作为一个点，有2n个点，如果第i位置不可以填j，将位置i向值j连边。

这样，就得到了一个二分图，问题就变成了选i条边的方案数。

将二分图的每条链拉出来，并在一起，就形成2n个点排成一排，一些相邻点之间有边。

设\(f[i][j][0/1]\)表示，做到第i个点，选了j条边，这个点与上个一点的边是否有选（如果没边就为0）的方案数。

那么\(g[i]=f[2n][i]][0]+f[2n][i][1]\)

#include <cmath>
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
#include <map>
#include <bitset>
#include <set>
const int inf=2147483647;
const int mo=924844033;
const int N=4005;
using namespace std;
int n,m,tot;
bool lk[N];
long long f[N][N][2],jc[N],ans;
int main()
{
	scanf("%d%d",&n,&m);
	jc[0]=1;
	for(int i=1;i<=n;i++) jc[i]=1ll*jc[i-1]*i%mo;
	for(int i=1;i<=m;i++)
		for(int t=2;t--;)
			for(int j=i;j<=n;j+=m)
			{
				tot++;
				if(j!=i) lk[tot]=1;
			}
	f[0][0][0]=1;
	for(int i=0;i<n*2;i++)
		for(int j=0;j<=n;j++)
		{
			f[i+1][j][0]=(f[i][j][0]+f[i][j][1])%mo;
			if(lk[i+1]) f[i+1][j+1][1]=f[i][j][0];
		}
	for(int i=0,t=1;i<=n;i++,t=-t)
	{
		f[2*n][i][0]=1ll*(f[2*n][i][0]+f[2*n][i][1])*jc[n-i]%mo;
		ans=(ans+f[2*n][i][0]*t+mo)%mo;
	}
	printf("%lld\n",ans);
}