B-number HDU - 3652（数位dp ）

题目描述

A wqb-number, or B-number for short, is a non-negative integer whose decimal form contains the sub- string “13” and can be divided by 13. For example, 130 and 2613 are wqb-numbers, but 143 and 2639 are not. Your task is to calculate how many wqb-numbers from 1 to n for a given integer n.
Input
Process till EOF. In each line, there is one positive integer n(1 <= n <= 1000000000).
Output
Print each answer in a single line.
Sample Input

13
100
200
1000

Sample Output

1
1
2
2

题目链接

• 题目大意：求从1到n中b数（能被十三整除并且含有“13”）的个数。
• 思路：根据题意，dp数组中有三个维度，dp[len][mod][status]代表长度为len,余数为mod，状态为status的数的个数。一共有以下三个状态：
  • 0：前面没有出现过“13”，并且上一位遍历到的不是1；
  • 1：前面没有出现过“13”，但是上一位遍历到的是1；
  • 2：前面已经出现过“13”。

对三种状态：

• 0：以前没有出现过“13”，看现在这一位是不是1，如果是1，那么下一位状态是1，否则是0；
• 1：上一位是1，看这一位是不是3，如果是，那么下一位状态为2，否则为0；
• 2：已经出现过“13”，下一位直接延续当前状态。

在遍历过程中同时取模，遍历完以后看余数是否为0即可。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int MAX_STATUS = 3, MAX_LEN = 10, MAX_MOD = 15;
int dp[MAX_LEN][MAX_MOD][MAX_STATUS], digit[MAX_LEN];

int Dfs(int len, int mod, int status, bool limit) {
    if(len == 0)    return status == 2 && mod == 0;
    if(limit == 0 && dp[len][mod][status] != -1)    return dp[len][mod][status];
    int ans = 0;
    int End = limit ? digit[len] : 9;
    for(int Digit = 0; Digit <= End; Digit++) {
        int mod_temp = (mod * 10 + Digit) % 13;
        if(status == 2 || (status == 1 && Digit == 3))
            ans += Dfs(len - 1, mod_temp, 2, limit && Digit == End);
        else if(Digit == 1)
            ans += Dfs(len - 1, mod_temp, 1, limit && Digit == End);
        else
            ans += Dfs(len - 1, mod_temp, 0, limit && Digit == End);
    }
    if(limit == 0)  dp[len][mod][status] = ans;
    return ans;
}

int Solve(int n) {
    int len = 0;
    while(n) {
        digit[++len] = n % 10;
        n /= 10;
    }
    return Dfs(len, 0, 0, true);
}

int main() {
    int n;
    memset(dp, -1, sizeof(dp));
    while(cin >> n)
        printf("%d\n", Solve(n));
    return 0;
}