Codeforces Round #518 (Div. 2): D. Array Without Local Maximums（DP）

有一个长度为n的序列，满足对于所有的a[x]，与它相邻的两个元素a[x-1]和a[x+1]中至少有一个大于等于它，其中a[1]和a[n]当然只有一个相邻元素， 现在这个序列中有些数字被破坏了（标记为-1），问有多少种合法恢复方案（每个数字∈[1,200]）

dp[2][100005][205]; 1当前数字ok么，2编号，3结尾数字，4有几个

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <string>
#include <queue>

using namespace std;

int n;
long long mod=998244353;

long long dp[2][100005][205];
int a[100005];

long long sum[2][205];
int main() {

    while(~scanf("%d",&n))
    {
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
        }
        if(a[1]==-1)
        {
            for(int i=1;i<=200;i++)
                dp[1][1][i]=1;
        }
        else
            dp[1][1][a[1]]=1;

        for(int i=2;i<=n;i++)
        {

            memset(sum,0,sizeof(sum));
            for(int j=1;j<=200;j++)
            {
                sum[0][j]=(sum[0][j-1]+dp[0][i-1][j])%mod;
                sum[1][j]=(sum[1][j-1]+dp[1][i-1][j])%mod;
            }
            int l=1,r=200;
            if(a[i]!=-1)
                l=a[i],r=a[i];

            for(int j=l;j<=r;j++)
            {
                dp[0][i][j]=(sum[0][200]-sum[0][j-1]+dp[1][i-1][j])%mod;
                dp[1][i][j]=(sum[0][j-1]+sum[1][j-1])%mod;
            }
        }
        long long ans=0;
        for(int i=1;i<=200;i++)
        {
            ans=(ans+dp[0][n][i])%mod;
        }
        ans=(ans+mod)%mod;
        printf("%lld\n",ans);
    }
    return 0;
}