题目大意
有一个 ( n + 2 ) ( n + 2 ) (n+2)(n+2) (n+2)(n+2)的网格,有一个排列 P n P_n Pn, ( i , P i ) (i,P_i) (i,Pi)是一个障碍,设 f ( P ) f(P) f(P)表示,不经过障碍的从 ( 0 , 0 ) (0,0) (0,0)走到 ( n + 1 , n + 1 ) (n+1,n+1) (n+1,n+1)的方案数, P P P的部分位置已经给出,求 ∑ f ( P ) \sum f(P) ∑f(P)
题解
考虑计算每一条路径,有多少种可能使得 P P P使得该路径不经过障碍,直接计算比较困难,想到容斥来弱化限制,如果我们能求出 g ( k ) g(k) g(k)表示有至少 k k k个障碍在路径上的方案数,那最后就可以求出最终的答案,如果要有 k k k个障碍在路径上,那他们对应的方位一定是递增的,所以就可以设 f i , j , k , 0 / 1 , 0 / 1 f_{i,j,k,0/1,0/1} fi,j,k,0/1,0/1表示路径走到 ( i , j ) (i,j) (i,j)已经经过至少 k k k个障碍,当前所在行是否有障碍,当前所在列是否有障碍的方案数,决策当前点是否放一个障碍,最后要给没有钦定位置的排列乘上一个阶乘
c o d e code code
#include<cstdio>
using namespace std;
const int N=3e2+10,mod=998244353;
int n,a[N+10],f[N+10][N+10][N+10][2][2],fac[N+10];
bool visa[N+10],visb[N+10],vis[N+10][N+10];
void add(int &a,int b){a+=b;if(a>=mod) a-=mod;}
int main()
{
scanf("%d",&n);int cnt=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
if(a[i]!=-1) visa[i]=true,visb[a[i]]=true,vis[i][a[i]]=true,cnt++;
}
f[0][0][0][0][0]=1;
for(int i=0;i<=n+1;i++)
for(int j=0;j<=n+1;j++)
{
if(vis[i][j]) continue;
for(int k=0;k<=n-cnt;k++)
{
add(f[i+1][j][k][0][0],f[i][j][k][0][0]),
add(f[i][j+1][k][0][0],f[i][j][k][0][0]),
add(f[i+1][j][k][0][1],f[i][j][k][0][1]),
add(f[i][j+1][k][0][0],f[i][j][k][0][1]),
add(f[i+1][j][k][0][0],f[i][j][k][1][0]),
add(f[i][j+1][k][1][0],f[i][j][k][1][0]);
}
if(!visa[i]&&!visb[j]&&i!=0&&j!=0&&i!=n+1&&j!=n+1)
for(int k=0;k<n-cnt;k++)
{
add(f[i+1][j][k+1][0][1],f[i][j][k][0][0]),
add(f[i][j+1][k+1][1][0],f[i][j][k][0][0]);
}
}
fac[0]=1;
for(int i=1;i<=n-cnt;i++) fac[i]=1ll*fac[i-1]*i%mod;
int ans=0;
for(int i=0;i<=n-cnt;i++)
{
if(i&1) add(ans,mod-1ll*fac[n-cnt-i]*f[n+1][n+1][i][0][0]%mod);
else add(ans,1ll*fac[n-cnt-i]*f[n+1][n+1][i][0][0]%mod);
}
printf("%d\n",ans);
return 0;
}