被3整除的子序列-DP

被3整除的子序列-DP

链接：https://ac.nowcoder.com/acm/problem/21302
来源：牛客网

题目描述

给你一个长度为50的数字串,问你有多少个子序列构成的数字可以被3整除
答案对1e9+7取模

输入描述:

输入一个字符串，由数字构成，长度小于等于50

输出描述:

输出一个整数

样例描述

输入

132

输出

3

输入

333

输出

7

输入

00

输出

3

数据范围

n为长度
子任务1: n <= 5
子任务2: n <= 20
子任务3: 无限制

思路：

核心：

一个数能够被3整除的条件：数字各位的和能够整除3.

因此，我们可以贪心的构造出使得一个序列数位之和能够满足整除3的情况。

倒着思考：

一个长度为len的序列，他的各个数位的之和等于长度为len-1的序列的各个数位的之和 + 第len位的数字的值。

一个长度为len-1的序列，他的各个数位之和等于长度为len-2的序列的各个数位的之和 + 第len-1位的数字的值。

…

一个长度为1的序列，他的各个数位的之等于长度为0的序列的各个数位的之和 + 第1位的数字的值。

一个长度为0的序列，他的各个数位的之为0。

抽象成表达式：

$dp[i][j]$：表示为前$i$位所能构成各个数位和为$j$的子序列数量

那么在第$i$个位置时，要构成的序列和为$j$的序列，有三种情况需要考虑，以下讨论划分为两种：

1. 在原本构成的序列的基础上，通过对第$i$位数字做出选与不选两种决策：

   选：$dp[i][j]=dp[i][j]+dp[i-1][j-num]$

   不选：$dp[i][j]=dp[i-1][j]$

   合并后：$dp[i][j]=dp[i-1][j]+dp[i-1][j-num]$

   满足各位数和$sum$能够被3整除，那么$sum\%3=0或1或2$

   由于$j-num$的结果可能为负值

   因此表达式$dp[i][j]=dp[i-1][j]+dp[i-1][(j-num+3)\%3]$

2. 还有一种情况不能漏掉，即以第$i$为起点的情况，

   此时的表达应为：$dp[i][num]=(dp[i][num]+1)\%mod$。

AC代码

#include<bits/stdc++.h>
using namespace std;
#define ios ios::sync_with_stdio(false)
#define tie cin.tie(0),cout.tie(0)
typedef long long ll;
const int INF = 0x7fffffff;
const int maxn = 65;
const int mod = 1e9+7;
int dp[60][3];
int t, n;

int main() {
    char s[maxn];
    cin >> (s+1);
    int len  = strlen(s+1);
    for(int i = 1; i <= len; i++){
        int num = (s[i] - '0') % 3;  //将输入的数处理在0，1，2这个范围中
        for(int j = 0; j < 3; j++){ 
            //j - num 可能为负数 因此加一个模数
            dp[i][j] = (dp[i-1][j] + dp[i-1][(j-num+3) % 3]) % mod;
        }
        //前i位构成的序列本身和为num
        dp[i][num] = (dp[i][num] + 1) % mod;
    }
    cout << dp[len][0] << endl;
    return 0;
}

贴上另外一种做法。

AC代码

#include<bits/stdc++.h>
using namespace std;
#define ios ios::sync_with_stdio(false)
#define tie cin.tie(0),cout.tie(0)
typedef long long ll;
const int INF = 0x7fffffff;
const int maxn = 100000 + 5;
const int mod = 1e9+7;
int a[60];
char s[60];
ll dp[60][2][3];  //前i位 选择或者不选择余数为0 1 2的数构成的子序列的个数

int main() {
    scanf("%s", s+1);
    int len = strlen(s+1);
    for(int i = 1; i <= len; i++){
        a[i] = (s[i] - '0') % 3;
    }
    // 一个数字能够被3整除的条件：数字各位和能够整除3
    dp[0][0][0] = 1;   //一个数都不选且余数为0的情况
    for(int i = 1; i <= len; i++){
        for(int j = 0; j < 3; j++){
            //初始状态下 余数为j的 不选择 直接继承上一个余数为j的状态的和
           dp[i][0][j] = dp[i-1][0][j] + dp[i-1][1][j];
           for(int k = 0; k < 3; k++){
               if((a[i] + k) % 3 == j){ //当前a[i]这个数字+k会生成一个新的序列和 如果加上这个序列和还是没变
                    //选择j这个余数来贡献序列和
                    dp[i][1][j] = (dp[i-1][0][k] + dp[i-1][1][k]) % mod;
               }
           }
        }
    }
    //最后的结果： 在len长度的序列出选择余数为0的个数 + 不选择余数为0的数 - 初始状态下的1 
    printf("%lld\n", (dp[len][1][0] + dp[len][0][0] - 1 + mod) % mod);
    return 0;
}