【贪心算法】leetcode402.移掉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。
思考
假设num = 1432219;k=1,分别去掉1,4,3,2,2,1,9得到数字: 分析:若去掉某一位数字,为了使得到的新数字最小,需要尽可能让新数字优先最高位最小,其次此高位最小,这样下去。。。。
贪心规律
从最高位向最低为遍历,如果对应的数字大于下一位数字,则把该数字丢掉,得到数字最小!暴力解法就是遍历K次,去掉K个数字。
解题思路&代码实现
在这里我们可以借助栈来实现,刚开始我也想到了但是各种细节调了俩小时,最终还是出来了。
public static String removeKdigits(String num, int k) { String ans = ""; Stack<Character> stack = new Stack<>(); for (int i = 0; i < num.length(); i++) { // k大于0,栈不为空,栈顶元素大于num【i】就可以删掉栈顶元素,这样才是最小的结果 while (k > 0 && !stack.isEmpty() && stack.peek() > num.charAt(i)) { stack.pop(); k--; } // 否则入栈 stack.push(num.charAt(i)); } // 有可能是12345,这样的,K大于0,直接从栈顶删就好了 while (!stack.isEmpty() && k-- > 0) { stack.pop(); } //拼接起来 while (!stack.isEmpty()) { ans = stack.pop() + ans; } // 处理一下前导0 int count = 0; for (int i = 0; i < ans.length(); i++) { if (ans.charAt(i) == 0) { count++; } else { break; } } ans = ans.substring(count) ; return ans.equals("")?"0":ans; }
流程:
public static String removeKdigits2(String num, int k) { String ans = ""; Stack<Integer> S = new Stack<>(); for (int i = 0; i < num.length(); i++) { int tmp = num.charAt(i) - 0; while (!S.isEmpty() && k > 0 && S.peek() > tmp) { S.pop(); k--; } if (S.isEmpty() || tmp != 0) { S.push(tmp); } } while (!S.isEmpty() && k-- > 0) { S.pop(); } while (!S.isEmpty()) { ans = S.pop() + ans; } return ans.equals("") ? "0" : ans; }