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;
    }
};
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容