题意:
给出n,k,问有多少个长度为n的排列满足
abs(ai−i)!=k
。
n<=2000
1<=k<=n-1
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<iostream>
#define N 2100
#define mmod 924844033
using namespace std;
int n,m,A[N],du[N],f[N],g[N][N][2],siz,num,p[N],jc[N],ans;
void upd(int &x,int y)
{
x=(x+y)%mmod;
}
void solve(int st)
{
num=0;
while(st<=n) {A[++num]=st;st+=2*m;}
for(int i=1;i<=num;i++)
{
du[i]=0;
if(A[i]-m>=1) du[i]++;
if(A[i]+m<=n) du[i]++;
}
if(num==1 && du[1]==0) {siz++;return;}
for(int i=1;i<=num;i++)
for(int j=0;j<=i;j++)
for(int o=0;o<=1;o++)
g[i][j][o]=0;
g[1][0][0]=1;g[1][1][1]=1;
if(du[1]==2) g[1][1][0]=1;
for(int i=1;i<num;i++)
for(int j=0;j<=i;j++)
{
upd(g[i+1][j][0],g[i][j][0]);
upd(g[i+1][j][0],g[i][j][1]);
upd(g[i+1][j+1][0],g[i][j][0]);
if(du[i+1]==2)
{
upd(g[i+1][j+1][1],g[i][j][0]);
upd(g[i+1][j+1][1],g[i][j][1]);
}
}
for(int i=0;i<=siz+num;i++) p[i]=0;
for(int i=0;i<=siz;i++)
for(int j=0;j<=num;j++)
upd(p[i+j],1ll*f[i]*(g[num][j][0]+g[num][j][1])%mmod);
siz+=num;
for(int i=0;i<=siz;i++) f[i]=p[i];
}
int main()
{
scanf("%d%d",&n,&m);
int lim=min(n,2*m);
siz=0;
f[0]=1;
for(int i=1;i<=lim;i++) solve(i);
jc[0]=jc[1]=1;
for(int i=2;i<=n;i++) jc[i]=1ll*jc[i-1]*i%mmod;
for(int i=0;i<=n;i++)
{
int t=1ll*jc[n-i]*f[i]%mmod;
if(i%2) upd(ans,-t);
else upd(ans,t);
}
upd(ans,mmod);
printf("%d\n",ans);
return 0;
}
题解:
考虑容斥,如果计算出i个位置填了错误的数,其他位置不填的方案数,那就容易计算答案。
然后发现这个图有些部分之间独立,具体来说互相影响的地方长这个样子:
把每部分拿出来独立dp,然后合并成全局方案数。合并过程相当于枚举每个点对,复杂度
O(n2)
题解里说的
nlogn
做法是什么呀qaq