当前位置: 代码迷 >> 综合 >> CF #518 (Div. 2) D. Array Without Local Maximums
  详细解决方案

CF #518 (Div. 2) D. Array Without Local Maximums

热度:77   发布时间:2023-12-12 00:22:29.0

传送门

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 012都可以加

等于的情况:只要把前面那个数的 0 , 1 , 2 0,1,2 012都加上。

大于的情况:加上大于这个数的数字方案数的前缀和,可以加上 1 , 2 1,2 12的情况。

空间开不下用滚动数组。

#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);
}
  相关解决方案