2021.8.26 栈:移掉K位数字(med)
题目:
给你一个以字符串表示的非负整数 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 <= 10⁵
num 仅由若干位数字(0 - 9)组成
除了 0 本身之外,num 不含任何前导零
Related Topics 栈 贪心 字符串 单调栈 👍 625 👎 0
思路:
因为要得到最小的数字,所以我们需要保证高位数字尽可能的小,因为高位变化1比低位变化从0-9更大。所以遍历时要将小的字符移动到前面去,利用栈来实现,即是入栈。
- 遍历字符串时当栈不为空,且当前字符ch小于栈顶字符,且能移掉的次数k大于0,我们就可以把满足此条件的所有栈顶元素出栈,然后k–,如果在这个while操作后,栈为空且ch为’0’,则不用将ch入栈,否则则将ch入栈。
- 遍历字符串结束后,如果k仍>0,则表示栈中有多余的元素,全部出栈
- 然后判断现在栈是否为空,为空直接返回0
- 栈不为空就拼接结果字符串,如果有前导0,则循环找到位置,返回位置之后的字符串,如果没有,判断结果字符串是否为"",返回0或结果字符串
代码:
class Solution {
public String removeKdigits(String num, int k) {
if (num.length()==k){
return "0";
}
Deque<Character> stack = new LinkedList<>();
for (int i = 0; i < num.length(); i++) {
char ch = num.charAt(i);
if (stack.isEmpty()){
stack.push(ch);
}else{
while (!stack.isEmpty()&&stack.peek()>ch&&k>0){
stack.pop();
k--;
}
if (stack.isEmpty()&&ch=='0'){
continue;
}
stack.push(ch);
}
}
while (k>0){
k--;
stack.pop();
}
if (stack.isEmpty()){
return "0";
}
StringBuilder sb = new StringBuilder();
while (!stack.isEmpty()){
sb.append(stack.pop());
}
String result = sb.reverse().toString();
for (int i = 0; i < result.length(); i++) {
if (result.charAt(i)!='0'){
return result.substring(i);
}
}
return result.equals("")?"0":result;
}
}