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

被3整除的子序列
给你一个长度为50的数字串,问你有多少个子序列构成的数字可以被3整除
答案对1e9+7取模
https://ac.nowcoder.com/acm/problem/21302

#include<iostream>
#include<string.h>
#define mod 1000000007
using namespace std;
int main()
{
	char str[100005];
	scanf("%s",str);
	int dp[10005][4];
	dp[0][(str[0]-'0')%3]=1;
	for(int i=1;i<=strlen(str);i++)
	{
		int temp=str[i]-'0';
		if(temp%3==0){//如果这个数mod3==0 
			dp[i][0]=(dp[i-1][0]+dp[i-1][0]+1)%mod;//不选这个数+选这个数,+1是说明单选这个数（遇到的j==temp%3就是不选这个数+选这个+单独选这个数） 
			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(temp%3==1){
			dp[i][0]=(dp[i-1][0]+dp[i-1][2])%mod;//不选这个数+选这个数，不选这个数的话就是求dp[i-1][0]，选这个数就是前i-1位中子序列各个元素的和mod3==2的个数
			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(temp%3==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;
		}
	}
//	for(int i=0;i<strlen(str);i++)
//	{
//		for(int j=0;j<=2;j++)
//		{
//			printf("%d ",dp[i][j]);
//		}
//		printf("\n");
//	}
	printf("%d\n",dp[strlen(str)-1][0]);
	return 0;
 }