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。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-k-digits

思路解析

以 [1,2,3,4,5,2,6,4] 为例：

要移除 1 位数字，那么应该移除 5，得到结果为 1234264
要移除 2 位数字，那么应该移除 5 和 4，得到结果为 123264
要移除 3 位数字，那么应该移除 5、4、3，得到结果为 12264
要移除 4 位数字，那么应该移除 5、4、3、6，得到结果为 1224

宗上可知，要移除的数字应该满足以下条件：

被移除的数字应该大于其后一位数字；
若多个数字满足情况，则须从前向后选择移除的数字；
每次移除一个数字后，应当更新整个字符串，并从前往后选取下一个要移除的数字。

为了加速，我们维护一个栈，栈自底向上以此递增，每一位数字在入栈之前，先与栈顶比较，若小于栈顶，则弹出栈顶，直至该数字大于栈顶，则如栈。
每次弹出栈顶的过程就是删除一个数字的过程。

最后，需要对得到的结果进行一定的处理，将栈转换为字符串后，需要删除字符串首的 "0"。

代码实现

class Solution {
public:
    string removeKdigits(string num, int k) {
        if(k == num.size()) return "0";
        stack<char> digitStk;
        for(auto digit : num) {
            while(!digitStk.empty() && (digit < digitStk.top()) && k > 0) {
                digitStk.pop();
                k--;
            }
            digitStk.push(digit);
        }
        while(!digitStk.empty() && k > 0) {
            digitStk.pop();
            k--;
        }
        if(digitStk.empty()) return "0";
        string result;
        while(!digitStk.empty()) {
            result = digitStk.top() + result;
            digitStk.pop();
        }
        int headIndex = 0;
        while(true) {
            if(result[headIndex] != '0') {
                break;
            }
            headIndex++;
        }
        result = result.substr(headIndex, result.size() - headIndex);
        result = (result.empty()) ? "0" : result;
        return result;
    }
};