leetcode402——Remove K Digits

题目大意:字符串num代表一个非负整数,删除其中k位使得数字变得最小,输出不能出现前导零和空串。

分析:单调栈+贪心。贪心策略:从左往右删,如果左邻居>当前位,就应该删掉左邻居,这样才能使数字变最小。栈:遍历字符串,当前位小于栈顶元素时弹栈,否则入栈。可以直接用字符串来实现这个单调栈。注意:1.可能遍历结束时还没删够k位,需要直接删掉末尾几位,也就是直接截取字符串的前size-k位。2.可能删除导致出现了前导零,比如10200,因此要删掉前导零再返回。3.当k>=size,会删除所有数字,此时需要返回“0”,而不是空串。

代码:

class Solution {
public:
    string removeKdigits(string num, int k) {
        string ans; //用字符串实现单调栈
        int m = num.size() - k; //num删除k位后的长度(ans的长度)
        for (int i = 0;i < num.size();i++) {
            while (k && ans.size() && ans.back() > num[i]) {
                ans.pop_back();
                k--;
            }
            ans += num[i];
        }
        ans.resize(m);
        while (ans.size() && ans[0] == '0') ans.erase(ans.begin());
        return ans.size() == 0 ? "0" : ans;
    }
};

 

你可能感兴趣的:(数据结构-栈)