传送门
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);
}