【牛客】被3整除的子序列（区间DP）

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

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 524288K，其他语言1048576K
64bit IO Format: %lld
题目描述
给你一个长度为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[i][j][0]表示i,j之间 子序列值%30 的子序列个数
dp[i][j][1]表示i,j之间 子序列值%31 的子序列个数
dp[i][j][2]表示i,j之间 子序列值%3==2 的子序列个数
则很容易得到状态转移方程：

s[j + 1] - ‘0’ % 3 == 0时：

•  		dp[i][j + 1][0] = 2*dp[i][j][0] + 1;
   		dp[i][j + 1][1] = 2*dp[i][j][1];
   		dp[i][j + 1][2] = 2*dp[i][j][2];
  

s[j + 1] - ‘0’ % 3 == 1时：

•  		dp[i][j + 1][0] = dp[i][j][0] + dp[i][j][2];
   		dp[i][j + 1][1] = dp[i][j][1] + dp[i][j][0] + 1;
   		dp[i][j + 1][2] = dp[i][j][1] + dp[i][j][2];
  

s[j + 1] - ‘0’ % 3 == 2时：

•  		dp[i][j + 1][0] = dp[i][j][0] + dp[i][j][1];
   		dp[i][j + 1][1] = dp[i][j][2] + dp[i][j][1];
   		dp[i][j + 1][2] = dp[i][j][0] + dp[i][j][2] + 1;
  

代码如下:

#include<bits/stdc++.h>

using namespace std;

const int mod = 1e9 + 7;

int dp[50][50][4];

int main(){
	string s;
	cin>>s;
	for(int i = s.length();i >= 1;i--){
		s[i] = s[i - 1];
	}
	for(int i = 1;i <= s.length();i++){
		dp[i][i][(s[i]-'0')%3]=1;
	}
	for(int i = 1;i <= s.length();i++){
		for(int j = i;j <= s.length();j++){
			if((s[j + 1]  - '0')% 3 == 0){
				dp[i][j + 1][0] = 2*dp[i][j][0] + 1;
				dp[i][j + 1][1] = 2*dp[i][j][1];
				dp[i][j + 1][2] = 2*dp[i][j][2];
			}else if((s[j + 1]  - '0')% 3 == 1){
				dp[i][j + 1][0] = dp[i][j][0] + dp[i][j][2];
				dp[i][j + 1][1] = dp[i][j][1] + dp[i][j][0] + 1;
				dp[i][j + 1][2] = dp[i][j][1] + dp[i][j][2];
			}else{
				dp[i][j + 1][0] = dp[i][j][0] + dp[i][j][1];
				dp[i][j + 1][1] = dp[i][j][2] + dp[i][j][1];
				dp[i][j + 1][2] = dp[i][j][0] + dp[i][j][2] + 1;
			}
			for(int k = 0;k < 3;k++){
				dp[i][j + 1][k] = dp[i][j + 1][k] % mod;
			}
		}
	}
	cout<<dp[1][s.length()][0];
	return 0;
}