链接:https://ac.nowcoder.com/acm/problem/21302
来源:牛客网
时间限制: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: 无限制
思想:
dp[i][j][0]表示i,j之间 子序列值%30 的子序列个数
dp[i][j][1]表示i,j之间 子序列值%31 的子序列个数
dp[i][j][2]表示i,j之间 子序列值%3==2 的子序列个数
则很容易得到状态转移方程:
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];
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][1] + dp[i][j][0] + 1; dp[i][j + 1][2] = dp[i][j][1] + dp[i][j][2];
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;
代码如下:
#include<bits/stdc++.h>
using namespace std;
const int mod = 1e9 + 7;
int dp[50][50][4];
int main(){
string s;
cin>>s;
for(int i = s.length();i >= 1;i--){
s[i] = s[i - 1];
}
for(int i = 1;i <= s.length();i++){
dp[i][i][(s[i]-'0')%3]=1;
}
for(int i = 1;i <= s.length();i++){
for(int j = i;j <= s.length();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][1] + dp[i][j][0] + 1;
dp[i][j + 1][2] = dp[i][j][1] + dp[i][j][2];
}else{
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;
}
}
}
cout<<dp[1][s.length()][0];
return 0;
}