leetcode 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。

解题思路

贪心策略：移除从字符串下标0开始的递增序列的最大值，依次移除K次，余下的即为答案，同时要进行异常情况处理，比如k大于等于字符串长度，最后移除k个字符后剩下都是0字符的情况。

代码（java）

class Solution {
    public String removeKdigits(String num, int k) {
        //移除序列中递增最大的数字
        if(num.length() <=k ) return "0";
        int i=0;
        for(i=0;i<k;i++){
            int removeIndex = getMaxIndex(num);
            if(removeIndex==0)num = num.substring(removeIndex+1,num.length());
            else if( removeIndex<num.length()-1)num = num.substring(0,removeIndex) + num.substring(removeIndex+1,num.length());
            else num = num.substring(0,removeIndex);
        }
        //判断是否结果全为0
        for(i=0;i<num.length();i++){
            if(num.charAt(i)!='0') break;
        }
        if(i==num.length())return "0";
        return num.substring(i,num.length());
    }
    public int getMaxIndex(String num){
        int index = 0,i=0;
        for(i=0;i<num.length()-1;i++){
            if(num.charAt(i)>num.charAt(i+1)){
                index = i;
                break;
            }
        }
        if(i==num.length()-1) index = num.length()-1;
        return index;
    }
}