数据结构与算法-栈-移掉K位数字

移掉K位数字

给你一个以字符串表示的非负整数 num 和一个整数 k ,移除这个数中的 k **位数字,使得剩下的数字最小。请你以字符串形式返回这个最小的数字。

示例 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 。

提示:

  • 1 <= k <= num.length <= 105
  • num 仅由若干位数字(0 - 9)组成
  • 除了 0 本身之外,num 不含任何前导零

思路:

数据结构与算法-栈-移掉K位数字_第1张图片
数据结构与算法-栈-移掉K位数字_第2张图片

代码:

class Solution {
    public String removeKdigits(String num, int k) {

        LinkedList<Character> stack = new LinkedList<>();

        for (int i = 0; i < num.length(); i++) {
            char c = num.charAt(i);
            while (k>0 && !stack.isEmpty() && stack.peekLast() > c) {
                stack.removeLast();
                k--;
            }
            stack.addLast(c);
        }

        while (k>0) {
            stack.removeLast();
            k--;
        }

        StringBuilder res = new StringBuilder();
        boolean leadingZero = true;
        while (!stack.isEmpty()) {
            Character digit = stack.pollFirst();
            if(digit =='0' && leadingZero) {
                continue;
            }
            leadingZero = false;
            res.append(digit);
        }

        return res.length()==0?"0":res.toString();

    }
}

你可能感兴趣的:(数据结构与算法,算法,java)