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

题目描述

给你一个长度为50的数字串,问你有多少个子序列构成的数字可以被3整除
答案对1e9+7取模

输入描述:

输入一个字符串，由数字构成，长度小于等于50

输出描述:

输出一个整数

---

dp[i][k]前i个数，凑出余数为k的方案

两种选择，一种从前面凑得转移过来，一种什么不做直接加上前面的方案

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int N=60;
int n,a[N],dp[N][3],mod=1e9+7;
signed main(){
    char c;
    while(1){c=getchar();if(c=='\n')break;a[++n]=c-'0';a[n]%=3;}
    for(int i=1;i<=n;i++){
        for(int k=0;k<3;k++)
        dp[i][k]=(dp[i][k]+dp[i-1][(k-a[i]+3)%3])%mod;
        for(int k=0;k<3;k++)
        dp[i][k]=(dp[i][k]+dp[i-1][k])%mod;
        dp[i][a[i]]++;
    }
    cout<<dp[n][0]%mod;
}