402. 移掉K位数字(贪心+单调栈)

package com.heu.wsq.leetcode;

import java.util.Arrays;

/**
 * 402. 移掉K位数字
 * @author wsq
 * @date 2020/12/10
 * 给定一个以字符串表示的非负整数 num，移除这个数中的 k 位数字，使得剩下的数字最小。
 *
 * 注意:
 *
 * num 的长度小于 10002 且 ≥ k。
 * num 不会包含任何前导零。
 * 示例 1 :
 *
 * 输入: num = "1432219", k = 3
 * 输出: "1219"
 * 解释: 移除掉三个数字 4, 3, 和 2 形成一个新的最小的数字 1219。
 * 示例 2 :
 *
 * 输入: num = "10200", k = 1
 * 输出: "200"
 * 解释: 移掉首位的 1 剩下的数字为 200. 注意输出不能有任何前导零。
 *
 * 链接：https://leetcode-cn.com/problems/remove-k-digits
 */
public class RemoveKdigits {
    /**
     * 贪心 + 单调栈
     * @param num
     * @param k
     * @return
     */
    public String removeKdigits(String num, int k) {
        int len = num.length();
        int remain = len - k;
        int[] stack = new int[remain];
        int top = -1;
        // 尽可能将小的往前移
        for (char c : num.toCharArray()) {
            int n = c - '0';
            while (k > 0 && top >= 0 && stack[top] > n){
                top--;
                k--;
            }
            if (top < remain - 1){
                stack[++top] = n;
            }else {
                k--;
            }
        }
        // 找到第一个不为0的下标值
        int pos = -1;
        for (int i = 0; i < stack.length; i++) {
            if (stack[i] != 0){
                pos = i;
                break;
            }
        }
        // 如果不能找到不为0的元素，直接返回"0"
        if(pos == -1){
            return "0";
        }
        // 从下标pos起，将数字拼为字符串
        StringBuilder sb = new StringBuilder();
        for(int i = pos; i < stack.length; i++){
            sb.append(stack[i]);
        }
        return "".equals(sb.toString()) ? "0" : sb.toString();
    }
}