题目大意
给定一个以字符串表示的非负整数 num,移除这个数中的 k 位数字,使得剩下的数字最小。
注意:
- num 的长度小于 10002 且 ≥ k。
- num 不会包含任何前导零。
示例 1 :
输入: num = "1432219", k = 3
输出: "1219"
解释: 移除掉三个数字 4, 3, 和 2 形成一个新的最小的数字 1219。
示例 2 :
输入: num = "10200", k = 1
输出: "200"
解释: 移掉首位的 1 剩下的数字为 200. 注意输出不能有任何前导零。
解题思路
流程:
使用一个单调递增栈,遍历当前字符串。如果栈不为空且K>0且栈顶元素大于当前元素,则不断将栈顶元素弹出,弹出过程停止后,如果栈为空且当前元素为’0’,则不进入栈中,否则将当前元素压入栈中。
为何使用递增站栈(或者说为何找到降序位置的元素后,将前一个元素删除掉):例如s=‘12519’,k=1。移除k位数字后,位数从5位变成了4位,降低了一位。我们找到降序位置,即’51’,删除’5’后,'1’向前移动了一位,相当于高位数字变低了,因此数字整体上变小了。
class Solution {
public:
string removeKdigits(string num, int k) {
if (num.size() <= k)
return "0";
stack<char> st;
for (int i = 0; i < num.size(); i++)
{
while (st.size() > 0 && k > 0 && st.top() > num[i])
{
st.pop();
--k;
}
if (st.size() == 0 && num[i] == '0')
continue;
st.push(num[i]);
}
while (k != 0)
{
st.pop();
--k;
}
if (st.size() == 0)
return "0";
string ans = "";
while (st.size() > 0)
{
ans = st.top() + ans;
st.pop();
}
return to_string(stoi(ans));
}
};