贪心法——LeetCode 402 移除K个数字

移除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次

使用栈，从高位向低位遍历num
前面的数字大于栈顶元素，则将该数字push放入栈；
前面的数字小于栈顶元素，则将该数字pop弹出栈；
直到栈为空或不能再删除数字（k==0）或栈顶小于当前元素为止

最后结果为1219

• C++代码

class Solution {
public:
    string removeKdigits(string num, int k) {
        std::vector<int> s;
        std::string result="";
        for(int i=0;i<num.length();i++){
            int number=num[i]-'0';  //字符型转化为整数型
            while(s.size()!=0 && s[s.size()-1]>number && k>0){    //栈不空，栈顶元素大于numbernumber，一直删
                s.pop_back(); //弹出元素
                k--;
            }
            if(number!=0 || s.size()!=0){  //考虑为0的情况
                s.push_back(number);
            }
        }
        while(s.size()!=0&&k>0){ 
            s.pop_back();
            k--;
        }
        for(int i=0;i<s.size();i++){
            result.append(1,'0'+s[i]);  //数字转字符
        }
        if(result==""){
            result="0"; //结果为空，为0
        }
        return result;
    }
};