【DP】Codeforces - 1067A - Array Without Local Maximums

---

题目链接<https://cn.vjudge.net/problem/1983591/origin>

---

题意：

有一个长度为N的序列（2≤n≤1e5）满足关系：

a1≤a2，an≤an−1，ai≤max(ai−1,ai+1)。

每一个ai的范围是[1，200]，有一些ai是确定的，有一些是不确定的，问有多少种情况。答案对998244353取模。

---

题解：

从头到尾扫一遍，枚举每一位放数字的情况。

开一个数组dp[i][j]（1≤i≤200，j={0,1,2}）表示当前位置放数字i，与前一位的大小情况：

• dp[i][0]表示当前位放i且比前一位数字小
• dp[i][1]表示当前位放i且和前一位数字一样
• dp[i][2]表示当前位放i且比前一位数字大

另外开一个数组sum[i][j]（1≤i≤200，j={0,1,2}）是上一位dp数组的前缀和。

然后两者之间互相转移，对于是-1和不是-1两种情况做不同的转移。

注意：

1. 中间步骤会爆int，最好直接开long long。
2. 前缀和因为做了一个取模操作，后面的不一定会比前面的大，所以前缀和做减法的时候需要加上一个mod。

---

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=1e5+7;
const ll mod=998244353;
ll a[N];
ll dp[300][3],sum[300][3];
int main(){
    ll n;
    scanf("%lld",&n);
    for(ll i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    if(a[1]==-1){
        for(ll i=1;i<=200;i++){
            dp[i][1]=1;
            sum[i][1]=(sum[i-1][1]+1)%mod;
        }
    }
    else{
        dp[a[1]][1]=1;
        for(ll i=a[1];i<=200;i++)
            sum[i][1]=1;
    }
    for(ll i=2;i<=n;i++){
        if(a[i]==-1){
            for(ll j=1;j<=200;j++){
                dp[j][1]=(dp[j][0]+dp[j][1]+dp[j][2])%mod;
                dp[j][0]=(sum[200][0]-sum[j][0]+sum[200][1]-sum[j][1]+mod)%mod;
                dp[j][2]=(sum[j-1][0]+sum[j-1][1]+sum[j-1][2])%mod;
                if(i==2) dp[j][0]=0;
            }
        }
        else{
            for(ll j=1;j<a[i];j++) dp[j][0]=dp[j][1]=dp[j][2]=0;
            for(ll j=a[i]+1;j<=200;j++) dp[j][0]=dp[j][1]=dp[j][2]=0;
            dp[a[i]][1]=(dp[a[i]][0]+dp[a[i]][1]+dp[a[i]][2])%mod;
            dp[a[i]][0]=(sum[200][0]-sum[a[i]][0]+sum[200][1]-sum[a[i]][1]+mod)%mod;
            dp[a[i]][2]=(sum[a[i]-1][0]+sum[a[i]-1][1]+sum[a[i]-1][2])%mod;
            if(i==2) dp[a[i]][0]=0;
        }
        for(ll j=1;j<=200;j++){
            sum[j][0]=(sum[j-1][0]+dp[j][0])%mod;
            sum[j][1]=(sum[j-1][1]+dp[j][1])%mod;
            sum[j][2]=(sum[j-1][2]+dp[j][2])%mod;
        }
    }
    if(a[n]==-1)
        printf("%lld\n",(sum[200][0]+sum[200][1])%mod);
    else
        printf("%lld\n",(dp[a[n]][0]+dp[a[n]][1])%mod);
}