牛客网 被3整除的子序列 dp

时间限制：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: 无限制

 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=55;
const int MOD=1e9+7;
char s[N];
int a[N],dp[N][N*10];//dp[i][j]表示前i个数各个位置的数加起来能组成j的方法数 
int main(){
	while(cin>>s){
		memset(dp,0,sizeof(dp)); 
		int len=strlen(s);
		for(int i=0;i<len;i++){
			a[i+1]=s[i]-'0';
		}
		for(int i=1;i<=len;i++){
			for(int j=0;j<=N*10;j++){
				dp[i][j]=dp[i-1][j];
				if(a[i]==j) dp[i][j]++;
				if(a[i]<=j) dp[i][j]+=dp[i-1][j-a[i]];
				dp[i][j]%=MOD;
			}
		}
		ll ans=0;
		for(int i=0;i<=N*10;i++){
			if(i%3==0){
				ans=(ans+dp[len][i])%MOD;
			}
		}
		cout<<ans<<endl;
	}
	return 0;
}