402. 移掉K位数字(一般)

给定一个以字符串表示的非负整数 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。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-k-digits

思路:

栈+贪心算法

先举个例子436,我们要求只删除一个元素,所以我们有4,3,6三个元素进行选择,那么我们从3开始向右选择,3<4所以去掉4,因为不这样做,后面得到的,如43,46,都会比36大,永远不会得到最小数,这里我们用栈接收,之后弹出后逆序了,需要逆置一下

代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
class Solution {
public String removeKdigits(String num, int k) {
if(k>=num.length() || num.length()==0)
return "0";
//栈,顶存最大值
Stack<Integer> stack = new Stack<>();
//先存第一个
stack.push(num.charAt(0)-'0');
for(int i=1;i<num.length();i++){
int curr = num.charAt(i)-'0';
//如果栈不空,或者当前这个数小于栈顶元素,或者k>0
while(!stack.isEmpty()&&curr<stack.peek()&&k>0){
stack.pop();
k--;
}
//不等于0,或者等于0或栈不为空
if(curr!=0 || !stack.isEmpty())
stack.push(curr);
}
//如果遍历完一遍没有删完k个数,就顺着把栈尾的数删去
//因为栈顶元素最大,所以按顺序删即可
while(k>0){
k--;
stack.pop();
}
if(stack.isEmpty())
return "0";
//组装String
StringBuilder sb = new StringBuilder();
while(!stack.isEmpty())
sb.append(stack.pop());
return sb.reverse().toString();
}
}