E Xieldy And His Password 西安电子科技大学第16届程序设计竞赛网络同步赛

Elliott__

于 2018-04-24 16:03:08 发布

阅读量213

点赞数

CC 4.0 BY-SA版权

分类专栏：动态规划

本文链接：https://blog.youkuaiyun.com/monochrome00/article/details/80065978

动态规划专栏收录该内容

36 篇文章

订阅专栏

本文解析了NowCoder平台上的一个动态规划题目，通过二维DP数组记录以i为结尾时，字符串中数字3的出现次数，实现了高效求解。

题目链接<https://www.nowcoder.com/acm/contest/107/E>

题解：

设立二维dp[i][j]表明以i为结尾，%3为j的个数，从头到尾扫描一遍进行更新。

#include <bits/stdc++.h>
typedef long long LL;
using namespace std;
int dp[1000005][3];
int main(){
    string s;
    while(cin>>s){
        memset(dp,0,sizeof(dp));
        int ls=s.length();
        LL ans=0;
        for(int i=0;i<ls;i++){
            if(s[i]=='0'){
                dp[i+1][0]=dp[i][0]+1;
                dp[i+1][1]=dp[i][2];
                dp[i+1][2]=dp[i][1];
            }
            else{
                dp[i+1][0]=dp[i][1];
                dp[i+1][1]=dp[i][0]+1;
                dp[i+1][2]=dp[i][2];
            }
            ans+=dp[i+1][0];
        }
        cout<<ans<<endl;
    }
}