链接:https://ac.nowcoder.com/acm/problem/21302
来源:牛客网
被3整除的子序列
给你一个长度为50的数字串,问你有多少个子序列构成的数字可以被3整除
答案对1e9+7取模
输入描述:
输入一个字符串,由数字构成,长度小于等于50
输出描述:
输出一个整数
示例1
输入
132
输出
3
#include<bits/stdc++.h>
using namespace std;
long long dp[55][55][5];
long long mod=1e9+7;
long long inf=0x3f3f3f3f;
char s[55];
int main()
{
scanf("%s",s+1);
s[0]='z';
int n=strlen(s)-1;
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
{
dp[i][i][(s[i]-'0')%3]=1;
}
for(int i=n;i>=1;i--)
{
for(int j=i;j<n;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][0]+dp[i][j][1]+1;
dp[i][j+1][2]=dp[i][j][1]+dp[i][j][2];
}
else if((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;
}
for(int k=0;k<3;k++)
{
dp[i][j+1][k]=dp[i][j+1][k]%mod;
}
}
}
printf("%lld\n",dp[1][n][0]);
}