【贪心算法】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;
    }