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();
    }
}
经验分享 程序员 微信小程序 职场和发展