Pinely Round 3 (Div. 1 + Div. 2) F2. Small Permutation Problem (Hard Version)（组合数学+数形结合）

题目

给一个n(n<=2e5)，再给一个长为n的数组a(-1<=ai<=n)，

ai有一些项是-1，代表对位置没有限制，其他的值则代表有限制

考虑那些有限制的位置，求n的所有排列p中，满足：

p的前i项中，小于等于i的项恰为ai(0<=ai<=n)的方案数，答案对998244353取模

实际t(t<=1e4)组样例，保证sumn不超过2e5

easy：所有位置都有限制

hard：有一些位置是-1，代表这些位置没有限制

思路来源

官方题解

题解

官方题解写的还是挺不错的，复制一下他这个图，翻译一下题解

注意先判掉非法情况，

1. a[n]既不为-1，也不为n的情况

2. a[i]>i的情况

3. 除去-1的部分，其他值需要满足非严格单调递增，即a[las]>a[i]的情况是非法的

把题目选点想象成一个n*n的图，排列中第i个数为pi时，视为选中了(i,pi)这个点（图中红圈）

那么，<=i的时候ai的限制，就可以看成是在左上角i*i的矩阵中，选了ai个点

因为有一些位置是-1，所以直接找到前一个限制的位置las，

初始时，可以令las=0，a[las]=0

上一个约束处是las，限制为a[las]

当前约束处是i，限制为a[i]

可以发现多了一个L形状的区域可以选点，这部分需要选v=a[i]-a[las]个点，

如图所示，其中，w1=i-las，h1=las-a[las]，h2=i-las，w2=i-a[las]

那么，我们枚举w1*h1这个矩阵中，选了k个点，这k个点选了之后，同行同列不能再选点

那么对于下面的部分就还剩h2*(w2-k)这个矩阵，需要选剩下的v-k个点

可以从0枚举到a[i]-a[las]枚举这个k，因为邻相差之和最终等于n，所以复杂度是O(n)的

对于一个h*w的矩阵，在其中选择k个点的方案是：C(h,k)*C(w,k)*k!

即先选择k个横坐标在哪，再选择k个纵坐标在哪，再将纵坐标顺序打乱拼到横坐标上

代码

// Problem: F2. Small Permutation Problem (Hard Version)
// Contest: Codeforces - Pinely Round 3 (Div. 1 + Div. 2)
// URL: https://codeforces.com/contest/1909/problem/F2
// Memory Limit: 256 MB
// Time Limit: 2000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define per(i,a,b) for(int i=(a);i>=(b);--i)
typedef long long ll;
typedef double db;
typedef pair<ll,int> P;
#define fi first
#define se second
#define pb push_back
#define dbg(x) cerr<<(#x)<<":"<<x<<" ";
#define dbg2(x) cerr<<(#x)<<":"<<x<<endl;
#define SZ(a) (int)(a.size())
#define sci(a) scanf("%d",&(a))
#define scll(a) scanf("%lld",&(a))
#define pt(a) printf("%d",a);
#define pte(a) printf("%d\n",a)
#define ptlle(a) printf("%lld\n",a)
#define debug(...) fprintf(stderr, __VA_ARGS__)
const int N=2e5+10,mod=998244353;
int t,n,a[N];
int Finv[N],fac[N],inv[N];
int modpow(int x,int n,int mod){
	int res=1;
	for(;n;x=1ll*x*x%mod,n>>=1)
	if(n&1)res=1ll*res*x%mod;
	return res;
}
void init(int n){ //n<N
    inv[1]=1;
    for(int i=2;i<=n;++i)inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
	fac[0]=Finv[0]=1;
	for(int i=1;i<=n;++i)fac[i]=1ll*fac[i-1]*i%mod,Finv[i]=1ll*Finv[i-1]*inv[i]%mod;
	//Finv[n]=modpow(fac[n],mod-2,mod);
	//for(int i=n-1;i>=1;--i)Finv[i]=1ll*Finv[i+1]*(i+1)%mod;
}
int C(int n,int m){
	if(m<0||m>n)return 0;
	return 1ll*fac[n]*Finv[n-m]%mod*Finv[m]%mod;
}
//h*w的矩形选k个数的方案数
int f(int h,int w,int k){
	//if(!k)return 1;
	return 1ll*C(h,k)*C(w,k)%mod*fac[k]%mod;
}
int sol(){
	sci(n);
	int las=0,ans=1;
	rep(i,1,n){
		sci(a[i]);
	}
	if(a[n]==-1)a[n]=n;
	if(a[n]!=n)return 0;
	rep(i,1,n){
		if(a[i]==-1)continue;
		if(a[i]>i || a[i]<a[las])return 0;
		int v=a[i]-a[las];
		int w1=i-las,h1=las-a[las],h2=i-las,w2=i-a[las];
		//printf("i:%d v:%d h1:%d w1:%d h2:%d w2:%d\n",i,v,h1,w1,h2,w2);
		int sum=0;
		rep(j,0,v){
			sum=(sum+1ll*f(h1,w1,j)*f(h2,w2-j,v-j)%mod)%mod;
		}
		ans=1ll*ans*sum%mod;
		las=i;
	}
	return ans;
}
int main(){
	init(2e5);
	sci(t); // t=1
	while(t--){
		pte(sol());
	}
	return 0;
}