【牛客-21302】被3整除的子序列(动态规划)
被3整除的子序列
题目描述
给你一个长度为50的数字串,问你有多少个子序列构成的数字可以被3整除
答案对1e9+7取模
输入描述:
输入一个字符串,由数字构成,长度小于等于50
输出描述:
输出一个整数
示例1
输入
132
输出
3
示例2
输入
9
输出
1
题目链接:
这题仔细想想,可以这样写,每个数字串都可以从第一个最后一个去考虑,设初始子序列个数为len,每增加一个数,子序列就变为2*len+1;
设dp[0],dp[1],dp[2],表示除3得到的余数分别为0,1,2。每增加一个数就更新一遍数组。
余数为0:dp[0]=dp[0]+dp[0]+1;
dp[1]=dp[1]+dp[1];
dp[2]=dp[2]+dp[2];
余数为1:dp[0]=dp[0]+dp[2];