Java?C++題解leetcode902最大為N的數字組合數位DP

更新時間：2022年10月19日 11:41:26 作者：AnjaVon

這篇文章主要為大家介紹了Java?C++題解leetcode902最大為N的數字組合數位DP，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪

題目要求

題目鏈接

閱讀理解

思路：數位DP

Java

class Solution {
    public int atMostNGivenDigitSet(String[] digits, int n) {
        // 轉存digits
        int[] nums = new int[digits.length];
        for (int i = 0; i < digits.length; i++)
            nums[i] = Integer.parseInt(digits[i]);
        // 轉存n
        List<Integer> list = new ArrayList<>();
        while (n != 0) {
            list.add(n % 10);
            n /= 10;
        }
        int len = list.size(), m = nums.length, res = 0;
        // 目標數位數 = len
        for (int i = len - 1, p = 1; i >= 0; i--, p++) {
            int cur = list.get(i);
            int l = 0, r = m - 1;
            while (l < r) { // 二分找合適的digits[r]
                int mid = l + r + 1 >> 1;
                if (nums[mid] <= cur)
                    l = mid;
                else
                    r = mid - 1;
            }
            // 是否繼續(xù)向后
            if (nums[r] > cur)
                break;
            else if (nums[r] == cur) {
                res += r * (int)Math.pow(m, (len - p));
                if (i == 0) // 構造至最后一位
                    res++; // 加上nums[r]做該位的可能
            }
            else if (nums[r] < cur) {
                res += (r + 1) * (int)Math.pow(m, (len - p));
                break;
            }
        }
        // 目標數位數 < len
        for (int i = len - 1; i > 0; i--)
            res += Math.pow(m, i);
        return res;
    }
}

時間復雜度：O(log?n)，由于二分最大范圍是1∼91可忽略，所以整體復雜度僅與n的位數有關
空間復雜度：O(C)，轉存給出數據

C++

class Solution {
public:
    int atMostNGivenDigitSet(vector<string>& digits, int n) {
        // 轉存digits
        vector<int> nums;
        for (int i = 0; i < digits.size(); i++)
            nums.emplace_back(stoi(digits[i]));
        // 轉存n
        vector<int> list;
        while (n != 0) {
            list.emplace_back(n % 10);
            n /= 10;
        }
        int len = list.size(), m = nums.size(), res = 0;
        // 目標數位數 = len
        for (int i = len - 1, p = 1; i >= 0; i--, p++) {
            int cur = list[i];
            int l = 0, r = m - 1;
            while (l < r) { // 二分找合適的digits[r]
                int mid = l + r + 1 >> 1;
                if (nums[mid] <= cur)
                    l = mid;
                else
                    r = mid - 1;
            }
            // 是否繼續(xù)向后
            if (nums[r] > cur)
                break;
            else if (nums[r] == cur) {
                res += r * (int)pow(m, (len - p));
                if (i == 0) // 構造至最后一位
                    res++; // 加上nums[r]做該位的可能
            }
            else if (nums[r] < cur) {
                res += (r + 1) * (int)pow(m, (len - p));
                break;
            }
        }
        // 目標數位數 < len
        for (int i = len - 1; i > 0; i--)
            res += pow(m, i);
        return res;
    }
};