ARC118E Avoid Permutations

题目大意

有一个$(n+2)(n+2)$的网格，有一个排列$P_n$，$(i,P_i)$是一个障碍，设$f(P)$表示，不经过障碍的从$(0,0)$走到$(n+1,n+1)$的方案数，$P$的部分位置已经给出，求$\sum f(P)$

题解

考虑计算每一条路径，有多少种可能使得$P$使得该路径不经过障碍，直接计算比较困难，想到容斥来弱化限制，如果我们能求出$g(k)$表示有至少$k$个障碍在路径上的方案数，那最后就可以求出最终的答案，如果要有$k$个障碍在路径上，那他们对应的方位一定是递增的，所以就可以设$f_{i,j,k,0/1,0/1}$表示路径走到$(i,j)$已经经过至少$k$个障碍，当前所在行是否有障碍，当前所在列是否有障碍的方案数，决策当前点是否放一个障碍，最后要给没有钦定位置的排列乘上一个阶乘

$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;
}