链接:https://ac.nowcoder.com/acm/problem/21302
来源:牛客网
题目描述
给你一个长度为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: 无限制
因为是可以不连续的子序列,所以最后要考虑的就是从整个序列中,能取出多少种数,它们的和为3的倍数。
观察数的和最大只有9*50=450,那么可以用背包的思想来做:
dp[i][j]表示前i个中组合,和为j的个数,那么可以用01背包的思路,对于每个数有选或不选两种,思路就很清晰了。
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <iostream>
#define int long long
using namespace std;
const int mod=1e9+7;
int dp[100][500];
char s[100];
int num[100];
signed main(){
scanf("%s",s+1);
int len=1;
while(s[len]!='\0')len++;
len--;
for(int i=1;i<=len;i++){
num[i]=s[i]-'0';
}
memset(dp,0,sizeof(dp));
for(int i=1;i<=len;i++){
for(int j=0;j<=450;j++){
dp[i][j]=dp[i-1][j];
if(j==num[i])dp[i][j]++;
if(j>=num[i]){
dp[i][j]+=dp[i-1][j-num[i]];
dp[i][j]%=mod;
}
}
}
int ans=0;
for(int i=0;i<=450;i+=3){
if(i%3==0){
ans+=dp[len][i];
ans%=mod;
}
}
printf("%lld\n",ans);
return 0;
}