牛客-被3整除的子序列（DP）

链接：https://ac.nowcoder.com/acm/problem/21302
来源：牛客网

被3整除的子序列

题目描述

给你一个长度为50的数字串,问你有多少个子序列构成的数字可以被3整除
答案对1e9+7取模
链接：https://ac.nowcoder.com/acm/problem/21302
来源：牛客网

示例1
输入
复制
132
输出
复制
3
示例2
输入
复制
9
输出
复制
1
示例3
输入
复制
333
输出
复制
7
示例4
输入
复制
123456
输出
复制
23
示例5
输入
复制
00
输出
复制
3

解题思路：

被三整除的数的一个性质是所有数位的数加起来是3的倍数。枚举三个状态，模3等于0，模3等于1和模3等于2，那么考虑第i个位置时，考虑3种可能，i不加入到前i-1的序列，和i加入到前i-1的序列，以及以i为起始位置。

AC代码：

#include <iostream>
#include <algorithm>
#define int long long

using namespace std;
const int N = 1e6+10;
const int mod = 1e9+7;
long long dp[55][5];


signed main()
{
    string s;
    cin>>s;
    int len = s.size();
    dp[0][0] = 0;dp[0][1] = 0;dp[0][2] = 0;
    dp[0][(s[0]-'0')%3] = 1;
    for(int i = 1 ; i < len ; i ++)
    {
        for(int j = 0 ; j < 3 ; j ++)
            dp[i][j] = (dp[i-1][j]+dp[i-1][(3+j-(s[i]-'0')%3)%3])%mod;
        dp[i][(s[i]-'0')%3] = (dp[i][(s[i]-'0')%3]+1)%mod;
    }
    cout<<dp[len-1][0]<<endl;
}