402. Remove K Digits

思路 从前往后遍历数字,如果某一位的数字比下一位大,说明只要删除这一位,让下一位数字来代替这一位,就可以让整个数字变小。

所以维护一个非递减栈,从前向后遍历每一位,循环把比当前位大的栈顶弹出,直到k的次数用尽,或者栈弹光了。再把当前位放进栈里面。 遍历以后,如果给的k足够大,栈里的元素从栈底到栈顶,应当是非递减排列的。 如果还有剩余的k的次数,就高位往低位删除数字(因为高位的数字比较大。同样位数的数字中,位数越低,就应当放置越小的数字,而把大的数字留在高位)。 最后删除前导“0”。 另外补充个细节:如果全部字符删除掉了,说明答案是0,就返回“0”。 代码

class Solution {
public:
    string removeKdigits(string num, int k) {
        stack<char> st;
        string str;
        for (int i = 0; i < num.size(); i++) {
            while (k > 0 && !st.empty() && num[i] < st.top()) {
                k--;
                st.pop();
            }
            
            st.push(num[i]);
        }
        
        stack<char> str_st;
        while (k--)
            st.pop();
        
        while (!st.empty()) {
            str_st.push(st.top());
            st.pop();
        }
        
        while (!str_st.empty() && str_st.top() == '0')
            str_st.pop();
        
        while (!str_st.empty()) {
            str += str_st.top();
            str_st.pop();
        }
        
        if (str == "")
            return "0";
        
        stringstream ss;
        ss << str;
        string res;
        ss >> res;
        return res;
    }
};

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券