思路
从前往后遍历数字,如果某一位的数字比下一位大,说明只要删除这一位,让下一位数字来代替这一位,就可以让整个数字变小。
所以维护一个非递减栈,从前向后遍历每一位,循环把比当前位大的栈顶弹出,直到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;
}
};