链接:https://ac.nowcoder.com/acm/problem/21302
来源:牛客网
题目描述
给你一个长度为50的数字串,问你有多少个子序列构成的数字可以被3整除
答案对1e9+7取模
输入描述:
输入一个字符串,由数字构成,长度小于等于50
输出描述:
输出一个整数
示例1
输入 132
输出 3
思路:
dp,记录每个位置中,除以3余数是1,2,3的序列个数
#include <iostream>
#include <vector>
#include <queue>
#include <string>
#include <unordered_map>
#include <stack>
#include <limits.h>
using namespace std;
const int mod = 1e9+7;
int process(string str){
int n = str.size();
if(n == 0) return 0;
vector<vector<int>> dp(n,vector<int>(3,0));
dp[0][(str[0]-'0')%3] = 1;
for(int i = 1;i < n;i++){
int cur = str[i]-'0';
// 当前是是一个个体
dp[i][cur%3] = 1;
for(int j = 0;j < 3;j++){
// 不加入当前数
dp[i][j] += dp[i-1][j];
// 加入当前数
dp[i][(cur+j)%3] += dp[i-1][j];
}
dp[i][0] %= mod;
dp[i][1] %= mod;
dp[i][2] %= mod;
}
return dp[n-1][0];
}
int main(){
string str;
cin >> str;
int res = process(str);
cout << res << endl;
}
第二种更新dp
#include <iostream>
#include <vector>
#include <queue>
#include <string>
#include <unordered_map>
#include <stack>
#include <limits.h>
using namespace std;
const int mod = 1e9+7;
int process(string str){
int n = str.size();
if(n == 0) return 0;
vector<vector<int>> dp(n,vector<int>(3,0));
dp[0][(str[0]-'0')%3] = 1;
for(int i = 1;i < n;i++){
int cur = str[i]-'0';
if(cur%3 == 0){
dp[i][0] = 2*dp[i-1][0]+1;
dp[i][1] = 2*dp[i-1][1];
dp[i][2] = 2*dp[i-1][2];
}
else if(cur%3 == 1){
dp[i][0] = dp[i-1][0]+dp[i-1][2];
dp[i][1] = dp[i-1][1]+dp[i-1][0]+1;
dp[i][2] = dp[i-1][2]+dp[i-1][1];
}
else{
dp[i][0] = dp[i-1][0]+dp[i-1][1];
dp[i][1] = dp[i-1][1]+dp[i-1][2];
dp[i][2] = dp[i-1][2]+dp[i-1][0]+1;
}
dp[i][0] %= mod;
dp[i][1] %= mod;
dp[i][2] %= mod;
}
return dp[n-1][0];
}
int main(){
string str;
cin >> str;
int res = process(str);
cout << res << endl;
}