给你一个以字符串表示的非负整数 num 和一个整数 k ,移除这个数中的 k 位数字,使得剩下的数字最小。请你以字符串形式返回这个最小的数字。
示例 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 。
提示:
1 <= k <= num.length <= 105
num 仅由若干位数字(0 - 9)组成
除了 0 本身之外,num 不含任何前导零
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-k-digits
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
本题简直要气死我,本来我就想着使用贪心,因为之前用 C++ 做过,翻了翻以前的代码,用 java 复刻了一下,但是发现死活通过不了,后来才发现以前的代码写错了…我心态炸裂,本题使用贪心算法,遍历一个字符串,如果是降序的,就删除开始降序的第一个数字,如图:
此时数组只剩下 172,再次遍历找到降序的,删除,如果是有序的,就开始从后往前删除数字,全都删除 k 位数字以后,开始删除开始的零,最后输出即可
此算法的时间复杂度很高,有更优的解法,需要复盘
class Solution {
public String removeKdigits(String num, int k) {
if (k == num.length()) {
return "0";
}
int len = num.length();
StringBuilder sb = new StringBuilder(num);
for (int i = 0; i < k; i++) {
int index = 0;
for (int j = 1; j < sb.length() && sb.charAt(j) >= sb.charAt(j - 1); j++) {
index = j;
}
sb.deleteCharAt(index);
}
while (sb.length() > 1 && sb.charAt(0) == '0') {
sb.deleteCharAt(0);
}
return sb.toString();
}
}
第二种方法,也是面试的时候推荐的方法,单调栈,思路如下:
我们想要拿到移除 k 为数字最小的数,从上一解题方法,我们已知,如果是递减的,那么就需要删除当前位置的数字,比如 143,4 -> 3 是递减的,所以删除 4,变成 13,那么完全可以借助一个辅助空间进行:假设 k 等于 3
此时,只删去了递减的 4,k 此时等于 2,还需要删除 2 个元素,所以从后往前删除,此时只剩了 1。
由于如果是栈,还需要拿出来,再颠倒一下位置,所以此题我们使用双端队列来模拟
class Solution {
public String removeKdigits(String num, int k) {
Deque<Character> deque = new LinkedList<>();
for (int i = 0; i < num.length(); i++) {
// 此次循环删除递减的元素
while (!deque.isEmpty() && k > 0 && deque.peekLast() > num.charAt(i)) {
deque.pollLast();
k--;
}
deque.offerLast(num.charAt(i));
}
// 此次循环删除 k 个元素
while (k > 0 && !deque.isEmpty()) {
deque.pollLast();
k--;
}
// 如果为空,直接返回
if (deque.size() == 0) {
return "0";
}
// 如果不为空,用 StringBuilder 进行接收并返回
StringBuilder sb = new StringBuilder();
while (!deque.isEmpty() && deque.peekFirst() == '0') {
deque.pollFirst();
}
while (!deque.isEmpty()) {
sb.append(deque.pollFirst());
}
return sb.toString().equals("") ? "0" : sb.toString();
}
}