Leetcode 402. 移掉K位数字 C++

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。

题解

单调栈
我们可以维持一个单调递增栈，也就是说有出现递减的情况，我们就需要删除数字。但因为限制了删除次数，因此我们只能在允许删除次数内维持递增栈。
另外，如果最后还剩余删除次数，我们就把低几位删掉。详细过程见代码

代码

string removeKdigits(string num, int k) {
        int n = num.length(),i=1;
        stack<char> sp;
        sp.push(num[0]);
        while(k!=0 && i<n){
            while(k!=0 && !sp.empty() && num[i]<sp.top()){		//维持递增栈，只要能删，我们就删掉递减的元素
                sp.pop();
                k--;
            }
            if(num[i]=='0' && sp.empty()) while(num[i] =='0') i++;		///避免前导0
            else sp.push(num[i++]);
        }
        if(k!=0 && k >= sp.size()) return "0";
        string ans="";
        while(!sp.empty()){
            ans += sp.top();
            sp.pop();
        }  
        ans = ans.substr(k);		//删掉低k位数
        reverse(ans.begin(),ans.end());
        ans += num.substr(i);
        if(ans == "")   ans = "0";
        return ans;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-k-digits
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。