f i j k f_{ijk} fijk 第 i i i个数为 j j j时, k = 0 , 1 , 2 分 别 表 示 k={0,1,2}分别表示 k=0,1,2分别表示它左边的数小于,等于,大于它的方案数。
小于的情况,与大于类似, 0 , 1 , 2 0,1,2 0,1,2都可以加
等于的情况:只要把前面那个数的 0 , 1 , 2 0,1,2 0,1,2都加上。
大于的情况:加上大于这个数的数字方案数的前缀和,可以加上 1 , 2 1,2 1,2的情况。
空间开不下用滚动数组。
#include<bits/stdc++.h>
using namespace std;
const int mo=998244353;
long long n,k,a[100005],f[2][205][3],ans,s;
int main(){
scanf("%lld",&n);
for(int i=1;i<=n;++i) scanf("%lld",a+i);
if(a[1]==-1) for(int i=1;i<=200;++i) f[k][i][0]=1;else f[k][a[1]][0]=1;
for(int i=2;i<=n;k^=1,++i){
s=0;
for(int j=1;j<=200;++j){
f[k^1][j][0]=(a[i]==-1||a[i]==j)?s:0;
(s+=f[k][j][0]+f[k][j][1]+f[k][j][2])%=mo;
}
for(int j=1;j<=200;++j)
f[k^1][j][1]=(a[i]==-1||a[i]==j)?(f[k][j][0]+f[k][j][1]+f[k][j][2])%mo:0;
s=0;
for(int j=200;j;--j){
f[k^1][j][2]=(a[i]==-1||a[i]==j)?s:0;
(s+=f[k][j][1]+f[k][j][2])%=mo;
}
}
for(int i=1;i<=200;++i) (ans+=f[k][i][1]+f[k][i][2])%=mo;
return !printf("%lld",ans);
}