Xieldy And His Password

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


101010
输出

5

思路：这是一道dp题。dp[i][j]表示以i结尾的，模为j的数的个数，状态转移方程由：

  dp[i][j]由dp[i-1][***]的情况进行转移；

因为当最后加上0是相当于原数*2

最后加上1是相当于原数*2+1；

于是得以下状态转移方程：

当新加的一位是0的时候:①dp[i][0]+=dp[i-1][0]；②dp[i][1]+=dp[i-1][1];③dp[i][2]+=dp[i-1][2];

当新加一位是1的时候：①dp[i][1]+=dp[i-1][0];②dp[i][2]+=dp[i-1][1];③：dp[i][0]+=dp[i-1][1];

代码：

#include <bits/stdc++.h>
using namespace std;
 
const int maxn=1e6+300;
char str[maxn];
long long dp[maxn][3];
int main()
{
    while(~scanf("%s",str))
    {
        memset(dp,0,sizeof(dp));
        int len = strlen(str);
        if(len == 0) 
        {
            printf("0\n");continue;
        }
        dp[0][str[0]-'0']=1;
        long long res = dp[0][0];
        for(int i=1;i<len;i++)
        {
            int f = str[i]-'0';
            dp[i][f]++;
            if(f==0)
            {
                dp[i][0]+=dp[i-1][0];
                dp[i][2]+=dp[i-1][1];
                dp[i][1]+=dp[i-1][2];
            }
            else
            {
                dp[i][1]+=dp[i-1][0];
                dp[i][0]+=dp[i-1][1];
                dp[i][2]+=dp[i-1][2];
            }
            res += dp[i][0];
        }
        printf("%lld\n",res);
    }
    return 0;
}