牛客专题——被3整除的子序列——背包DP

链接：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;

}