CF 1067A

CF 1067A DP(算是数位DP? 算是第一道做出来的数位DP 嘻嘻)
题目链接:http://codeforces.com/contest/1067/problem/A

题意:
有一个长度为N的序列(2≤n≤1e5)满足:

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

每一个ai的范围是[1,200]

求有多少种满足条件的序列
答案对998244353取模

dp[maxn][200][3] 根据每位与前一位关系转移
dp 第三维为0 表示 小于前一位
为1 表示 大于前一位
为2 表示 等于前一位

在转移过程中用前缀和来求区间和
因为空间不够 开滚动数组(其实dp数组也可以改成滚动,但是嫌麻烦就没改)
因为要取模,前缀和部分有可能减除负数,因此对答案加mod取模

#include<bits/stdc++.h>
#define mod 998244353
using namespace std;
long long dp[100005][205][3];
long long sum[2][205][3];
int a[100005];
int main(){
    int n;
    long long r=0;
    int pre=0;
    int now=1;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=200;i++){
        dp[0][i][0]=1;
        dp[0][i][1]=1;
        dp[0][i][2]=1;
    }
    if(a[1]==-1){
        for(int i=1;i<=200;i++){
            dp[1][i][0]=0;
            dp[1][i][1]=1;
            dp[1][i][2]=0;
            sum[pre][i][0]=0;
            sum[pre][i][1]=sum[pre][i-1][1]+1;
            sum[pre][i][2]=0;
        }
    }
    else{
        for(int i=1;i<=200;i++){
            if(i==a[1]){
                dp[1][i][0]=0;
                dp[1][i][1]=1;
                dp[1][i][2]=0;
                sum[pre][i][0]=0;
                sum[pre][i][1]=sum[pre][i-1][1]+1;
                sum[pre][i][2]=0;
            }
            else{
                dp[1][i][0]=0;
                dp[1][i][1]=0;
                dp[1][i][2]=0;
                sum[pre][i][0]=0;
                sum[pre][i][1]=sum[pre][i-1][1];
                sum[pre][i][2]=0;
            }
        }
    }
    for(int i=2;i<n;i++){
        for(int j=1;j<=200;j++){
            if(a[i]==-1){
                dp[i][j][0]=(sum[pre][200][0]-sum[pre][j][0]+sum[pre][200][2]-sum[pre][j][2])%mod;
                dp[i][j][1]=(sum[pre][j-1][1]-sum[pre][0][1]+sum[pre][j-1][0]-sum[pre][0][0]+sum[pre][j-1][2]-sum[pre][0][2])%mod;
                dp[i][j][2]=(dp[i-1][j][2]+dp[i-1][j][0]+dp[i-1][j][1])%mod;
                sum[now][j][0]=(sum[now][j-1][0]+dp[i][j][0])%mod;
                sum[now][j][1]=(sum[now][j-1][1]+dp[i][j][1])%mod;
                sum[now][j][2]=(sum[now][j-1][2]+dp[i][j][2])%mod;
            }
            else{
                if(j==a[i]){
                    dp[i][j][0]=(sum[pre][200][0]-sum[pre][j][0]+sum[pre][200][2]-sum[pre][j][2])%mod;
                    dp[i][j][1]=(sum[pre][j-1][1]-sum[pre][0][1]+sum[pre][j-1][0]-sum[pre][0][0]+sum[pre][j-1][2]-sum[pre][0][2])%mod;
                    dp[i][j][2]=(dp[i-1][j][2]+dp[i-1][j][0]+dp[i-1][j][1])%mod;
                    sum[now][j][0]=(sum[now][j-1][0]+dp[i][j][0])%mod;
                    sum[now][j][1]=(sum[now][j-1][1]+dp[i][j][1])%mod;
                    sum[now][j][2]=(sum[now][j-1][2]+dp[i][j][2])%mod;
                }
                else{
                    dp[i][j][0]=0;
                    dp[i][j][1]=0;
                    dp[i][j][2]=0;
                    sum[now][j][0]=sum[now][j-1][0];
                    sum[now][j][1]=sum[now][j-1][1];
                    sum[now][j][2]=sum[now][j-1][2];
                }
            }
        }
        for(int j=1;j<=200;j++){
            sum[pre][j][0]=sum[now][j][0];
            sum[pre][j][1]=sum[now][j][1];
            sum[pre][j][2]=sum[now][j][2];
            sum[now][j][0]=0;
            sum[now][j][1]=0;
            sum[now][j][2]=0;
        }
    }
    if(a[n]==-1){
    for(int i=1;i<=200;i++){
        dp[n][i][0]=(sum[pre][200][0]-sum[pre][i][0]+sum[pre][200][2]-sum[pre][i][2])%mod;
        dp[n][i][1]=0;
        dp[n][i][2]=(dp[n-1][i][2]+dp[n-1][i][0]+dp[n-1][i][1])%mod;
    }
    }
    else{
        for(int i=1;i<=200;i++){
            if(a[n]==i){
                dp[n][i][1]=0;
                dp[n][i][0]=(sum[pre][200][0]-sum[pre][i][0]+sum[pre][200][2]-sum[pre][i][2])%mod;
                dp[n][i][2]=(dp[n-1][i][2]+dp[n-1][i][0]+dp[n-1][i][1])%mod;
            }
            else{
                dp[n][i][1]=0;
                dp[n][i][0]=0;
                dp[n][i][2]=0;
            }
        }
    }
    for(int i=1;i<=200;i++){
        r=(r+dp[n][i][0])%mod;
        r=(r+dp[n][i][1])%mod;
        r=(r+dp[n][i][2])%mod;
    }
    cout<<(r+mod)%mod<<endl;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值