题目链接
思路:
要使得移掉K位数字后,剩下的数字最小,需要从头开始移除掉最大的K个元素,这K个元素都满足:比后面一位元素大,即在原始数字中呈现的是逆序状态,需要将其删除
使用单调栈来进行删除,栈内满足栈底到栈顶从小到大排列
从头开始遍历所给数字
1)若当前数字比栈顶数字大或等于,则压入栈中;
2)若当前数字比栈顶数字小,则出现了逆序,需要将栈顶元素弹出,直至栈顶元素小于等于当前数字为止,弹出的过程,也就是删除元素的过程,一共需要删除K个元素,所以当一共弹出了K个元素后,就不需要再弹出任何元素了,将剩余数字都压入栈中
最后,将栈中元素弹出进行拼接,并取掉前导零,即可得到最终结果
public String removeKdigits(String num, int k) {
if (num.length() == k) {
return "0";
}
Deque<Character> stack = new ArrayDeque<>();
char[] numChar = num.toCharArray();
for (int i = 0; i < numChar.length; i++) {
// 还需要弹出元素时,并且当前数字比栈顶数字小时,进行弹出
while (!stack.isEmpty() && numChar[i] < stack.peek() && k > 0) {
stack.pop();
k--;
}
// 将当前数字压入栈内
stack.push(numChar[i]);
}
// 若所给数字无逆序,或逆序的个数不够K个,则需要将最后面的弹出,直至一共弹出K个
while (k > 0 && !stack.isEmpty()) {
stack.pop();
k--;
}
// 对栈内剩余数字进行拼接
StringBuilder sb = new StringBuilder();
while (!stack.isEmpty()) {
// 从栈底开始往后拼接,或者可以从栈顶开始往前插入
sb.append(stack.pollLast());
}
// 取掉前导0
int trimIndex = 0;
while (trimIndex < sb.length()) {
if (sb.charAt(trimIndex) == '0') {
trimIndex++;
} else {
break;
}
}
if (trimIndex == sb.length()) {
return "0";
}
return sb.substring(trimIndex, sb.length());
}