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次
使用栈存储最终结果或删除工作,从高位向地位遍历num,如果遍历的数字大于栈顶元素,则
将数字push入栈,如果小于栈顶元素则进行pop弹栈,直到栈为空或不能再删除数字(k==0)或栈顶元素的值小于当前前元素的值。最后,从栈底遍历到栈顶
思考如下问题:
1、当所有数字都扫描完成后,k仍然大于0,应该做怎样的处理?
例如:num = 12345, k = 3时
2、当数字中有0出现时,应该有怎样的特殊处理?
例如:num = 100200, k = 1时
3、如何将最后结果存储为字符串并返回?
代码和解释如下:

class Solution {
public:
    string removeKdigits(string num, int k) {
        vector<int> s;   //使用vector当做栈(因为vector可以遍历)
        string result = "";  //储存最终结果的字符串
        for(int i = 0; i < num.length(); i++){  //从最高位循环扫描数字num
            int number = num[i] - '0';   //将字符型的num转化为整数使用
            while(s.size() != 0 && number < s[s.size()-1] && k > 0){
                s.pop_back();
                k--;
            }
            if(number != 0 || s.size() != 0){
                s.push_back(number);  //将数字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";   //如果result为空,result即为0
        }
        return result;
    }
};
发布了65 篇原创文章 · 获赞 4 · 访问量 1014

猜你喜欢

转载自blog.csdn.net/CLZHIT/article/details/103658639