402.移掉K位数字

402.移掉K位数字

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

思路

• 由于高位数字影响较大，我们应该尽量删除高位较大的数字，如526，删除5的效果永远比删除2的效果要好，因此可以贪心地遍历字符串，找出第一个位置，使得num[i] < num[i - 1]，此时就可以删除i - 1位置字符(i > 0)，需要删除k个数字就贪心遍历k次，但这个方案的时间复杂度达到O（nk）。
• 其实我们只要保证字符串是单调不降就好了，单调不降，很容易就想到单调递增栈，只要当前入栈的元素比栈顶元素小，就不断删除栈顶元素，直到碰见不大于该元素，或者栈为空，或者k为0。
• 处理完毕后，如果k还不为0,证明栈是单调不降，从后面删除剩余的k个数字即可。
• 此时对栈元素进行处理，有前导零就跳过，栈底到栈顶的元素就是最小数字。

代码

class Solution {
public:
    string removeKdigits(string num, int k) {
        deque<char> stk;
        for(auto& digit : num) {
            while(!stk.empty() && stk.back() > digit && k > 0) {
                stk.pop_back();
                k--;
            }
            stk.push_back(digit);
        }
        while(k > 0) {
            stk.pop_back();
            k--;
        }
        string ans = "";
        bool prefixzero = true;
        for(auto& digit : stk) {
            if(prefixzero == true && digit == '0') {
                continue;
            }
            prefixzero = false;
            ans += digit;
        }
        //字符串为空，返回0
        return ans == ""? "0" : ans;
    }
};