LeetCode902最大为 N 的数字组合(相关话题：数位DP问题，递归遍历和减枝)-CSDN博客

本文链接：https://blog.csdn.net/lzhcoder/article/details/132805850

本文详细解析了LeetCode902题的解决方案，通过递归遍历和减枝、动态规划两种方法，重点介绍了数位动态规划（DP）问题的思路，包括通用方法论、暴力统计、二维和一维数位DP的实现，并提供了相关代码模板。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

给定一个按 非递减顺序 排列的数字数组 digits 。你可以用任意次数 digits[i] 来写的数字。例如，如果 digits = ['1','3','5']，我们可以写数字，如 '13', '551', 和 '1351315'。

返回 可以生成的小于或等于给定整数 n 的正整数的个数 。

示例 1：

输入：digits = ["1","3","5","7"], n = 100
输出：20
解释：
可写出的 20 个数字是：
1, 3, 5, 7, 11, 13, 15, 17, 31, 33, 35, 37, 51, 53, 55, 57, 71, 73, 75, 77.

示例 2：

输入：digits = ["1","4","9"], n = 1000000000
输出：29523
解释：
我们可以写 3 个一位数字，9 个两位数字，27 个三位数字，
81 个四位数字，243 个五位数字，729 个六位数字，
2187 个七位数字，6561 个八位数字和 19683 个九位数字。
总共，可以使用D中的数字写出 29523 个整数。

示例 3:

输入：digits = ["7"], n = 8
输出：1

提示：

1 <= digits.length <= 9
digits[i].length == 1
digits[i] 是从 '1' 到 '9' 的数
digits 中的所有值都不同
digits 按 非递减顺序 排列
1 <= n <= 109

方法一递归遍历和减枝

每次计算不同长度的的数据，只有长度和n的长度相同时需要特殊考虑，其余情况通过全排列的公式可以直接计算

public class LeetCode902最大为N的数字组合 {


    public int[][] mem;

    public int atMostNGivenDigitSet(String[] digits, int n) {


        int result = 0;

        int length = String.valueOf(n).length();

        //分别缓存有高位限制和无高位限制下的情况
        mem = new int[length][2];

        Boolean maxLength = false;
        for (int i = 1; i <= length; i++) {

            if (length  == i) {
                maxLength = true;
            }
            result += getLessNByBitLength(digits, n, i, maxLength);
        }

        return result;
    }


    /**
     *
     * @param digits
     * @param n
     * @param length 对应n的长度，每次递归都减去1代表，已经确定了一位
     * @param maxLength 是否属于计算和n相同位数的递归(是否有高位限制)
     * @return
     */
    private int getLessNByBitLength(String[] digits, int n, int length, Boolean maxLength) {

        int flag = maxLength==true ?1:0;

        if (mem[length - 1][flag] != 0) {
            return mem[length - 1][flag];
        }

        String strN = String.valueOf(n);

        int result = 0;

        if (maxLength) {
            if(length == 1){
                int temp=0;
                for (int i = 0; i < digits.length; i++) {
                    if(strN.charAt(strN.length()-length) >= digits[i].charAt(0)){
                        temp++;
                    }

                }
                return temp;
            }

        }else {
            if (length == 1) return digits.length;
        }



        for (int i = 0; i < digits.length; i++) {

            if (maxLength) {
                if(strN.charAt(strN.length()-length) > digits[i].charAt(0)){
                    //不受高位限制影响，直接结算结果
                    result += Math.pow(digits.length, length - 1);
                }else if(strN.charAt(strN.length()-length) == digits[i].charAt(0)){
                     //高位限制影响，递归计算结果
                    result += getLessNByBitLength(digits,n,length-1,maxLength);
                }
            } else {
                //无高位限制直接结算结果
                result += Math.pow(digits.length, length - 1);
            }

        }

        mem[length - 1][flag] = result;

        return result;
    }