D - Array Without Local Maximums （dp）

题目链接

题意：

有一个长度为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;
}