牛客动态规划题1

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

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

复制
132
输出

复制
3
示例2
输入

复制
9
输出

复制
1
示例3
输入

复制
333
输出

复制
7
示例4
输入

复制
123456
输出

复制
23
示例5
输入

复制
00
输出

复制
3
备注:
n为长度
子任务1: n <= 5
子任务2: n <= 20
子任务3: 无限制

区间DP？应该算是吧，看别人博客这么说的，咱也不知道，咱也不敢问。。。

直接贴代码吧

#include<iostream>
#include<string.h>
#include<string>
using namespace std;
long long mod=1e9+7;
long long b[60][10];
int main ()
{
	
	int a[100];
	string s;
	cin>>s;
	for(int i=0;i<s.size();i++)
    {
	 	a[i]=s[i]-'0';
    }
	long long sum=0;
	
	for (int i = 0; i < s.size(); i++) {
		b[i][(s[i] - '0') % 3] = 1;
	}
	for(int i=1;i<s.size();i++)
	{
		if(a[i]%3==0)
		{
			b[i][0]=2*b[i-1][0]+1;
			b[i][1]=2*b[i-1][1];
			b[i][2]=2*b[i-1][2];
		}else if(a[i]%3==1)
		{
			b[i][0]=b[i-1][0]+b[i-1][2];
			b[i][1]=b[i-1][0]+b[i-1][1]+1;
			b[i][2]=b[i-1][1]+b[i-1][2];
			
		}else if(a[i]%3==2)
		{
			b[i][0]=b[i-1][0]+b[i-1][1];
			b[i][1]=b[i-1][2]+b[i-1][1];
			b[i][2]=b[i-1][0]+b[i-1][2]+1;
			
		}
		for (int j = 0; j < 3; j++) 
		{
			b[i][j] %= mod;
		}

		
	} 
	
	
	cout<<b[s.size()-1][0]<<endl;
	
	
	
}

代码不怎么复杂，但是理解起来还是有点困难的。
B数组用来储存当前数字个数时，余数分别为0,1,2的子串个数。
首先遍历一遍，把存在余数为0,1,2的b【0】【 】变为1，然后就是重点了。
它的状态转移方程有三种情况，新加进来的数可能余数为0,1,2。
当余数为0时，B[i]【0】是上一次的个数乘以2加上1，例如原来是3,6,9，现在加进来一个3，它和原来的任意一种组合都可以被3整除，所以要乘以2,再加上它本身，b[i][1】和3的任意组合还是1，所以要乘以2，同理b【i】【2】。
当新加进来的余数为1时，它和余数为2的子串任意组合都可以被3整除，所以b【i】【0】=b【i-1】【0】+b【i-1】【2】，
它和余数为零的子串任意组合余数都为1，所以b[i][1]=b[i-1][0]+b[i-1][1]+1，它和余数为1的组合余数变为2，所以b[i][2]=b[i-1][1]+b[i-1][2]。
同理可得新加余数为2时，b[i][0]=b[i-1][0]+b[i-1][1];b[i][1]=b[i-1][2]+b[i-1][1];b[i][2]=b[i-1][0]+b[i-1][2]+1。
我自己理解的大概就是这样了，如果有不对的还请大佬指点。
话说这是两颗星的题？这评分对菜鸡一点也不友好，手动愤怒！