面试题 17.06. 2出现的次数

题解：. - 力扣（LeetCode）. - 力扣（LeetCode）

数位 DP 通用模板_哔哩哔哩_bilibili

class Solution {
public:
    int numberOf2sInRange(int n) {
        std::string str = to_string(n);
        int len = str.size();
        std::vector<std::vector<int>> memo(len, std::vector<int>(len, -1));
        int begin = 0;
        // is_limit表示当前每个位置是否是按照n对应位置去，往下走的
        bool is_limit = true;
        // 表示当前是否为一个数字, 前导0的情况就不是一个数字
        bool is_num = false;
        // 记录出现2的次数
        int count = 0;
        return dfs(str, begin, is_num, is_limit, count, memo);
    }
private:
    int dfs(std::string& str, int begin, bool is_num, bool is_limit, int count,
            std::vector<std::vector<int>>& memo) {
        // 已经走动末尾
        if (begin >= str.size()) {
            // 是一个数字
            if (is_num) {
                return count;
            }
            return 0;
        }
        // 记忆话搜素
        if (!is_limit && is_num && memo[begin][count] != -1) {
            return memo[begin][count];
        }
        int result = 0;
        if (!is_num) {
            // 不是数字，可以选择跳过，is_limit就没有限制了, is_num也不是一个数字
            result += dfs(str, begin+1, false, false, count, memo);
        }
        // 判断当前位置可以选择数字的最大值
        // 如果被限制了，就只能是当前str位置的元素，没限制就是9
        int up = is_limit ? str[begin] - '0' : 9;
        for (int i = 1 - is_num; i <= up; ++i) {
            // 往下递归，且记录信息
            result += dfs(str, begin+1, true, is_limit && (i == up ? true : false),
                count + (i == 2 ? 1 : 0), memo);
        }
        // 更新记忆话,当前位置，前门出现了2的个数count个，后面可能的情况
        if (!is_limit) {
            memo[begin][count] = result;
        }
        // 返回
        return result;
    }
};