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不为0,我们继续将栈顶元素弹出。

最后需要注意去掉前导0。

class Solution {
public:
    string removeKdigits(string num, int k) {
        stack<char>stk;
        for(auto x:num){
            while(stk.size()&&stk.top()>x&&k){
                stk.pop();
                k--;
            }
            stk.push(x);
        }
        while(k--)stk.pop();
        string res;
        while(stk.size()){
            res+=stk.top();
            stk.pop();
        }
        reverse(res.begin(),res.end());
        int i=0;
        while(i<res.size()&&res[i]=='0')i++;
        if(i==res.size())return "0";
        return res.substr(i);
    }
};

 

posted @ 2019-07-08 20:45  YF-1994  阅读(267)  评论(0编辑  收藏  举报