被3整除的子序列 （牛客2星题）（线性DP）

 

题目：

给你一个长度为50的数字串,问你有多少个子序列构成的数字可以被3整除（输入一个字符串，由数字构成，长度小于等于50）
答案对1e9+7取模

样例1：输入：132  输出：3

样例2：输入：123456 输出：23

首先要知道被三整除的数字的特征   只要所有数位数字的和可以被三整出即可。

因此可以把字符串每个数字取模3 处理成 只有0 1 2三个字符的串。

二维数组dp[i] [j] 表示 前 i 个长度的所有子序列数位数字和取模 3 为 j 的数目（ 0 <= j <= 2)

有状态转移方程 ：dp[i] [j] = (dp[i - 1] [j] + dp[i - 1] [ (3 + j - m) % 3]) % 1000000007;

表示dp[i] [j]可以由两种情况递推过来：

1.  第 i 个数字不用， 直接继承 dp[i - 1] [j]。

2. 第 i 个数字与前边的一些数位和为 （ 3 +  j - m) % 3 的所有子序列相接。

最后这个数字自己构成长度为 1 的子序列单独加上。

比如样例132来说

i          j	0	1	2
1	0	1	0
2	1	2	0
3	3	2	2

#include<iostream>
#include<string.h>
using namespace std;
int main()
{
    string a;
    int dp[55][4],mod=1e9+7;
    cin>>a;
    memset(dp,0,sizeof(dp));
    dp[1][(a[0]-'0')%3]=1;
    for(int i=2;i<=a.length();i++)
    {
        int m=(a[i-1]-'0')%3;
        for(int j=0;j<=2;j++)
            dp[i][j]=(dp[i-1][j]+dp[i-1][(3+j-m)%3])%mod;
        dp[i][m]++;
    }
    printf("%d",dp[a.length()][0]%mod);
    return 0;
}

wdnmd我DP为啥学的这么差，看半天才明白