题目链接
题意:
有一个长度为N的序列(2≤n≤1e5)满足关系:
a1≤a2,an≤an−1,ai≤max(ai−1,ai+1)。
每一个ai的范围是[1,200],有一些ai是确定的,有一些是不确定的,问有多少种情况。答案对998244353取模。
相当于说是对于任意三个相邻的三个数a[i-1],a[i],a[i+1],只排出了a[i]>a[i-1],a[i]>a[i+1]这个情况,也就是一个山峰的样子
dp[i][j][k],考虑第i个的时候放置j的时候与第i-1个关系是k(0,1,2-- < = >)的方案数
特殊的就是当k=2,也就是说当前这个小于i-1,那么注意第i-1个不能是小于i-2个了(山峰)
前缀和优化,否则超时
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 998244353;
ll dp[100005][205][3],sum;///0< 1= 2>
int a[100005], n, i, j;
int main(){
cin>>n;
for(i=1;i<=n;i++)
cin>>a[i];
for(i=1;i<=200;i++)
dp[1][i][0] = (a[1] == -1 || a[1] == i) ? 1:0;///dp[i][j][k] ,number of (the i-th ,lay j ,cmp is k)
for(i=2;i<=n;i++){
for(j=1;j<=200;j++)///left == [i]
dp[i][j][1] = (a[i]==-1 || a[i]==j)?(dp[i-1][j][0]+dp[i-1][j][1]+dp[i-1][j][2])%mod:0;
sum = 0;
for(j=1;j<=200;j++){///left < [i]
dp[i][j][0] = (a[i]==-1 || a[i]==j)?sum:0;
sum = (sum+dp[i-1][j][0]+dp[i-1][j][1]+dp[i-1][j][2])%mod;
}
sum = 0;
for(j=200;j;j--){///left > [i],
dp[i][j][2] = (a[i]==-1 || a[i]==j)?sum:0;
sum = (sum+dp[i-1][j][1]+dp[i-1][j][2])%mod;
}
}
sum = 0;
for(i=1;i<=200;i++)
sum = (sum+dp[n][i][1]+dp[n][i][2])%mod;
cout<<sum<<endl;
return 0;
}