题目
链接:https://leetcode-cn.com/problems/remove-k-digits
给定一个以字符串表示的非负整数 num,移除这个数中的 k 位数字,使得剩下的数字最小。
注意:
num 的长度小于 10002 且 ≥ k。
num 不会包含任何前导零。
示例 1 :
输入: num = "1432219", k = 3
输出: "1219"
解释: 移除掉三个数字 4, 3, 和 2 形成一个新的最小的数字 1219。
示例 2 :
输入: num = "10200", k = 1
输出: "200"
解释: 移掉首位的 1 剩下的数字为 200. 注意输出不能有任何前导零。
示例 3 :
输入: num = "10", k = 2
输出: "0"
解释: 从原数字移除所有的数字,剩余为空就是0。
贪心 + 单调栈
思路:
两个长度相同得字符串1axxx
与1bxxx
,他们的大小取决于最左边字符的大小,因此如果a>b
则1axxx > 1bxxx
。因此要使删除字符后字符串尽可能小,则要保持靠前的数字尽可能小。
我们可以使用栈来维持这种单调性,也就是说,如果当前数字大于栈顶元素则将栈顶元素弹出,直到:
- 栈为空
- 删除了k个数
- 栈顶数字不大于当前数字
经过栈处理后,可能没有删除k个数字,也就是说会出现字符串单调不减的情况,如num= "12345" k = 1
,此时需要删除最后的k个数字就能保证删除后数字保持最小
由于栈后进先出的特性,因此得到结果后还需进行反转,为了避免翻转的操作可以使用一个双端队列来代替栈。
在最后处理结果序列时要注意删除前导"0"
,如果结果序列大小为0,则直接返回"0"
。
class Solution {
public String removeKdigits(String num, int k) {
// 使用双端队列最后返回结果时不用反转
Deque<Character> deque = new LinkedList<>();
// 队列得长度
int length = num.length();
for (int i = 0; i < length; i ++) {
char digit = num.charAt(i);
while (!deque.isEmpty() && k > 0 && deque.peekLast() > digit) {
deque.pollLast();
k--;
}
deque.offerLast(digit);
}
// 删除未删除得数字,如 num = "9"这种情况
for (int i = 0; i < k; i++) {
deque.pollLast();
}
// 得到答案,要注意处理前导0
StringBuilder ret = new StringBuilder();
boolean leadingZero = true;
while (!deque.isEmpty()) {
char digit = deque.pollFirst();
if (leadingZero && digit == '0') {
continue;
}
leadingZero = false;
ret.append(digit);
}
return ret.length() == 0 ? "0" : ret.toString();
}
}