单调栈系列-Leetcode 402. 移掉K位数字

问题描述

给定一个以字符串表示的非负整数 num，移除这个数中的 k 位数字，使得剩下的数字最小。

注意:

num 的长度小于 10002 且 ≥ k。
num 不会包含任何前导零。
示例 1 :

输入: num = “1432219”, k = 3
输出: “1219”
解释: 移除掉三个数字 4, 3, 和 2 形成一个新的最小的数字 1219。

示例 2 :

输入: num = “10200”, k = 1
输出: “200”
解释: 移掉首位的 1 剩下的数字为 200. 注意输出不能有任何前导零。

示例 3 :

输入: num = “10”, k = 2
输出: “0”
解释: 从原数字移除所有的数字，剩余为空就是0。

解题报告

为了使得剩下的数字最小，最好的策略是尽可能让高位的数字小一些。
所以维护一个 单调递增栈，当前元素小于栈顶元素时，将栈顶元素出栈直到当前元素大于等于栈顶元素【当然这个过程需要保证出栈的元素个数小于等于 k 】。

最后将栈中元素取出【当然需要保证删除的元素达到 k，如果当前删除的元素小于 k，则依次将栈中元素弹出，直到达到 k】。

另外需要注意一点是，由于在栈底的是数字的高位，所以在栈为空时，如果即将入栈的元素为 0，那么跳过入栈操作。

实现代码

class Solution{
    public:
        string removeKdigits(string num, int k) {
            stack<char> s;
            for (int i = 0; i < num.size(); i++)
            {
                while (!s.empty() && s.top() > num[i] && k)
                {
                    s.pop();
                    k--;
                }
                if (s.empty() && num[i] == '0')
                    continue;//跳过前置0
                s.push(num[i]);
            }
            string result;
            while (!s.empty())
            {
                if (k > 0)//当还要再移除数字的时候：从此时单调递增栈的top部删去数字
                    k--;
                else if (k == 0)//当不用再移除数字的时候：把字符串取出来到result
                    result += s.top();

                s.pop();	
            }
            reverse(result.begin(), result.end());//stl中的reverse函数
            return result == "" ? "0" : result;
        }
};

单调栈其实不一定要新创建一个栈，比如下面这种实现。

class Solution {
    public:
        string removeKdigits(string num, int k){
            string result;
            for (int i = 0; i < num.size(); i++){
                while (result.size() && k&&result.back() > num[i]){
                    result.pop_back();
                    k--;
                }
                if (result.size() == 0 && num[i] == '0')
                    continue;
                result+=num[i];
            }
            // 注意：这里必须同时满足!result.empty()，因为有可能因为前置0的原因而跳过某些元素。前置0放到数字开头是没有用途的。
            while (k > 0 && !result.empty()){
                result.pop_back();
                k--;
            }
            return result == "" ? "0" : result;
        }
};

${}^{[2]}$

参考资料

[1] Leetcode 402. 移掉K位数字
[2] 题解区： Smart_Shelly