被3整除的子序列 ———牛客竞赛 DP

链接：https://acm.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[N][M] 表示前N个字符的组合中和模3余M的组合的个数

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
ll dp[55][5],a[55];
char s[55];
int main()
{
	int n;
	scanf("%s",s+1);
	n=strlen(s+1);
	for(int i=1;i<=n;i++) a[i]=s[i]-'0';
	int m=a[1]%3;
	dp[1][m]=1;
	for(int i=2;i<=n;i++)
	{
		m=a[i]%3;
		if(m==0)
		{
			dp[i][0]=(dp[i-1][0]+dp[i-1][0]+1)%mod;
			dp[i][1]=(dp[i-1][1]+dp[i-1][1])%mod;
			dp[i][2]=(dp[i-1][2]+dp[i-1][2])%mod;
		}
		if(m==1)
		{
			dp[i][0]=(dp[i-1][0]+dp[i-1][2])%mod;
			dp[i][1]=(dp[i-1][1]+dp[i-1][0]+1)%mod;
			dp[i][2]=(dp[i-1][2]+dp[i-1][1])%mod;
		}
		if(m==2)
		{
			dp[i][0]=(dp[i-1][0]+dp[i-1][1])%mod;
			dp[i][1]=(dp[i-1][1]+dp[i-1][2])%mod;
			dp[i][2]=(dp[i-1][2]+dp[i-1][0]+1)%mod;
		}
	}
	printf("%lld\n",dp[n][0]);
	return 0;
}