365天挑战LeetCode1000题——Day 117 数位DP I

最新推荐文章于 2025-07-04 11:54:47 发布

原创最新推荐文章于 2025-07-04 11:54:47 发布 · 233 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #c++ #动态规划

LeetCode千题之路专栏收录该内容

124 篇文章

订阅专栏

这篇博客介绍了一个C++实现的算法，用于找到给定数字集合中最大的N位数组合。通过动态规划和回溯策略，算法首先计算出在不考虑顺序的情况下可能的数字组合数量，然后通过栈来构造有序的数字序列，从而找到最大值。博客内容涵盖了代码实现、思路解析以及关键函数的详细说明。

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

902. 最大为 N 的数字组合

在这里插入图片描述

代码实现

class Solution {
private:
    int solve(stack<int>& st, vector<int>& digits) {
        if (st.empty()) return 0;
        int tmp = st.top();
        st.pop();
        int pos = lower_bound(digits.begin(), digits.end(), tmp) - digits.begin();
        // cout << pos * int(pow(digits.size(), st.size())) << endl;
        if (pos < digits.size() && digits[pos] == tmp) {
            if (st.empty()) return pos + 1;
            return pos * int(pow(digits.size(), st.size())) + solve(st, digits);
        }
        return pos * int(pow(digits.size(), st.size()));
    }
public:
    int atMostNGivenDigitSet(vector<string>& digits, int n) {
        int ans = 0;
        int sz = digits.size();
        int i = 1;
        for (long long k = 1; k * 10 <= n; k *= 10) {
            ans += int(pow(sz, i));
            i++;
        }
        int tmp = n;
        sort(digits.begin(), digits.end());
        vector<int> digits_new;
        for (auto digit : digits) {
            digits_new.push_back(digit[0] - '0');
        }
        stack<int> st;
        while (tmp) {
            int digit = tmp % 10;
            st.push(digit);
            tmp /= 10;
        }
        return ans + solve(st, digits_new);
    }
};