leetcode402(移掉K位数字:贪心算法)
给定一个以字符串表示的非负整数 num,移除这个数中的 k 位数字,使得剩下的数字最小。 输入: num = “1432219”, k = 3 输出: “1219” 解释: 移除掉三个数字 4, 3, 和 2 形成一个新的最小的数字 1219。
题解:贪心算法,对于任意一个数,一定是越靠前的数字越小,整个数字越小,为了让移除K位数字后的数尽量的小,我们可以采用贪心算法,遍历原字符串num,当num[ i ]<=num[ i+1 ]时,指针后移,当num[ i ]>num[ i+1 ]时,移除num[ i ]以及num[ i ]之前比num[ i+1 ]大的数,直到移除的数的总数量为k或num[ i ]之前所有数字都被移除,在继续后移指针,重复上述操作,直到K个数字都被移除完。如果遍历完num后,依然没有移除完k个数,则从末尾往前移除数字,直到移除的数字个数为K为止。为了优化算法,我们可以用到栈来存储当前已经遍历到的所有数字。
class Solution { public String removeKdigits(String num, int k) { Deque<Character> deque = new LinkedList<Character>(); int length = num.length(); for (int i = 0; i < length; ++i) { char digit = num.charAt(i); while (!deque.isEmpty() && k > 0 && deque.peekLast() > digit) { deque.pollLast(); k--; } deque.offerLast(digit); } for (int i = 0; i < k; ++i) { deque.pollLast(); } StringBuilder ret = new StringBuilder(); boolean leadingZero = true; while (!deque.isEmpty()) { char digit = deque.pollFirst(); if (leadingZero && digit == 0) { continue; } leadingZero = false; ret.append(digit); } return ret.length() == 0 ? "0" : ret.toString(); } }