LeetCode

移除K个数字

2018-03-13  本文已影响3人  徐凯_xp

LeetCode 402. Remove K Digits
已知一个使用字符串表示的非负整数num,将num中的k个数字移除, 求移除k个数字后,可以获得的最小的可能的新数字。(num不会以0开头,num长度小于10002)

输入 : num = “1432219” , k = 3
在去掉3个数字后得到的很多很多可能里,如1432、4322、2219、1219、 1229...;去掉数字4、3、2得到的1219最小!

class Solution{
public:
    std::string removeKdigits(std::string num, int k){// 返回的结果用字符串来表示
    }
};

分析

假设 num = 1432219 ; k = 1, 分别去掉1, 4, 3, 2, 2, 1, 9 得到数字:



分析与常识:
若去掉某一位数字,为了使得到的新数字最小,需要尽可能让得到的新数字优先 最高位最小,其次次高位最小, 再其次第3位最小...

算法设计

使用栈存储最终结果或删除工作,从高位向低位遍历num,
如果遍历的数字大于栈顶元素,则将该数字push入栈
如果小于栈顶元素则进行pop弹栈,直到栈为空或不能再删除数字
(k==0)或栈顶小于当前元素为止。
最终栈中从栈底到栈顶存储的数字,即为结果。




边界条件

1.当所有数字都扫描完成后,k仍然>0,应该做怎样的处理? 例如: num = 12345, k = 3 时。
2.当数字中有0出现时,应该有怎样的特殊处理? 例如: num = 100200, k = 1 时。
3.如何将最后结果存储为字符串并返回?

class Solution{
public:
    std::string removeKdigits(std::string num, int k){
    std::vector<int> S;//使用vector当作栈(因为vector可以遍历)
    std::string result = "";存储最终结果的字符串
    for(int i = 0;i <num.length();i++){//最高位循环扫描数字num
        int number = num[i] - '0';//将字符型num转换整数;
        while(S.size() != 0 && S[S.size() -1]>number  && k>0){//当栈不空,栈顶元素大于number
            S.pop_back();
            k --;
        }
        if(number !=0 || S.size() != 0){
            S.push_back(number);
        }
    while(S.size() && k>0){//如果栈不空仍然可以删除数字
        S.pop_back();
        k--;
    for(int i = 0;i<S.size();i++){
        result.append(1,'0'+S[i]);//转换成字符串
      }
    if(result == ""){
          result = "0";
      }
    return result;
      
    }
};
上一篇下一篇

猜你喜欢

热点阅读