题目
402. 移掉 K 位数字
给你一个以字符串表示的非负整数 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 不含任何前导零
解法
public String removeKdigits(String num, int k) {
//维护一个单调递增的单调栈,该栈的大小是len(num)-k的长度
Deque<Character> stk = new ArrayDeque<>();
for (int i = 0; i < num.length(); i++) {
char c = num.charAt(i);
while (!stk.isEmpty() && stk.peek() > c && k > 0) {
stk.pop();
k--;
}
stk.push(c);
}
//防止k没有完全消耗完
/*num="9"
k=1
exprected:"0"
*/
while (k-- > 0) stk.pop();
StringBuilder sb = new StringBuilder();
while (!stk.isEmpty()) sb.append(stk.pop());
boolean headZero = true;//前导零
StringBuilder res = new StringBuilder();
for (char c : sb.reverse().toString().toCharArray()) {
if (c == '0' && headZero) continue;
res.append(c);
headZero = false;
}
return res.toString().equals("") ? "0" : res.toString();
}
另
public String removeKdigits(String num, int k) {
Deque<Character> stk = new ArrayDeque<>();
char[] ch = num.toCharArray();
for (int i = 0; i < ch.length; i++) {
while (!stk.isEmpty() && stk.peek() > ch[i] && k > 0) {
stk.pop();
k--;
}
stk.push(ch[i]);
}
while (k-- > 0) stk.pop();
StringBuilder sb = new StringBuilder();
while (!stk.isEmpty()) sb.append(stk.pop());
StringBuilder res = new StringBuilder();
boolean headZero = true;
for (char c : sb.reverse().toString().toCharArray()) {
if (c == '0' && headZero) {
continue;
}
res.append(c);
headZero = false;
}
return res.toString().equals("") ? "0" : res.toString();
}