牛客网-西安电子科技大学校赛-E

题目描述

众所周知，Xieldy最常用的口令是******。
为了改变这一现状，他random了一个01串，并从中截取了一段作为自己的口令。
他选择的口令满足以下条件：
1. 口令串表示的二进制数在十进制下可以被表示为3k(k>=0)。
2. 口令串可以有前导零。
现已经random出了01串，他想知道有多少种口令方案可以选择（不同的子段即为不同）。

输入描述:

若干组数据，每组数据仅一行01串s，表示random出来的的串，|s|<=1e6。

输出描述:

输出口令的方案数。

示例1

输入

101010

输出

5

dp[1000005][3];///表示前i个串中余数为j的个数；

0+'0'->0;

1+'0'->2;

2+'0'->1;

0+'1'->1;

1+'1'->0;

2+'1'->2;

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
#include <string>
#include <vector>
#include <queue>
#include <cmath>
#include <set>
#include <vector>

using namespace std;
char str[1000000];
int dp[1000005][3];///表示前i个串中余数为j的个数
int main()
{
    while(~scanf("%s",str+1))
    {
        long long ans=0;
        int len=strlen(str+1);
        dp[0][0]=dp[0][1]=dp[0][2]=0;
        for(int i=1;i<=len;i++)
        {
            if(str[i]=='0')
            {
                dp[i][0]=dp[i-1][0]+1;
                dp[i][1]=dp[i-1][2];
                dp[i][2]=dp[i-1][1];
            }
            else
            {
                dp[i][1]=dp[i-1][0]+1;  ///第i位为1可以单独为一段，因此要加1
                dp[i][0]=dp[i-1][1];
                dp[i][2]=dp[i-1][2];
            }
            ans+=dp[i][0];
        }
        cout<<ans<<endl;
    }

	return 0;
}