402. 移掉K位数字-中等-贪心算法

402. 移掉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。

题解

我们的思路就是：

• 从左到右遍历
• 对于遍历到的元素，我们选择保留。
• 若遍历到的元素比前面的元素大，则丢弃前面相邻的元素。

所以我们遍历k次num，每次删除一个元素

class Solution {
    public String removeKdigits(String num, int k) {
        if(k == num.length()) return "0";
        StringBuffer s = new StringBuffer(num);
        
        for(int i=0; i<k; ++i){
            boolean flag = true;
            for(int j=1; j<s.length(); ++j){
                if(s.charAt(j) < s.charAt(j-1)){
                    s.deleteCharAt(j-1);
                    flag = false;
                    break;
                } 
            }
            if(flag){
                s.deleteCharAt(s.length()-1);
            }
        }
        while(s.length() > 1 && s.charAt(0) == '0'){
            s.deleteCharAt(0);
        }
        return s.toString();
    }
}