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'; while(!stack.isEmpty()&&curr<stack.peek()&&k>0){ stack.pop(); k--; } if(curr!=0 || !stack.isEmpty()) stack.push(curr); } while(k>0){ k--; stack.pop(); } if(stack.isEmpty()) return "0"; StringBuilder sb = new StringBuilder(); while(!stack.isEmpty()) sb.append(stack.pop()); return sb.reverse().toString(); } }
|