给定一个以字符串表示的非负整数 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。
解题思路:
当前一个数字大于后一个数字时,就说明此时的数字不符合。我们得到的字符串一定是前缀为单调递增的字符串。
// 速度很慢
class Solution {
public String removeKdigits(String num, int k) {
Stack<Character> stack = new Stack<>();
for(char digit : num.toCharArray()) {
// 判断当值是否小于栈顶元素
while(!stack.isEmpty() && k > 0 && stack.peek() > digit) {
stack.pop();
k--;
}
if(stack.isEmpty() && digit == '0')
continue;
stack.push(digit);
}
// 如果字符串就是单调递增的,弹出栈顶k个元素
while(k > 0 && !stack.isEmpty()) {
stack.pop();
k--;
}
if(stack.isEmpty()) {
return "0";
}
StringBuilder sb = new StringBuilder();
boolean flag = true;
while(!stack.isEmpty()) {
sb.append(stack.pop());
}
return sb.reverse().toString();
}
}
// 官方题解
class Solution {
public String removeKdigits(String num, int k) {
LinkedList<Character> stack = new LinkedList<Character>();
for(char digit : num.toCharArray()) {
while(stack.size() > 0 && k > 0 && stack.peekLast() > digit) {
stack.removeLast();
k -= 1;
}
stack.addLast(digit);
}
/* remove the remaining digits from the tail. */
for(int i=0; i<k; ++i) {
stack.removeLast();
}
// build the final string, while removing the leading zeros.
StringBuilder ret = new StringBuilder();
boolean leadingZero = true;
for(char digit: stack) {
if(leadingZero && digit == '0') continue;
leadingZero = false;
ret.append(digit);
}
/* return the final string */
if (ret.length() == 0) return "0";
return ret.toString();
}
}
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/remove-k-digits 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。