402. 移掉 K 位数字-贪心

Posted 2022-04-25 hequnwang10

一、题目描述

给你一个以字符串表示的非负整数 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 。

二、解题

贪心

这题主要是对于一些特殊情况的处理，首先是对字符串长度的一些判断。
然后根据K值，使用一个新的字符数组保存返回值：

• 如果后面的值小于前面的一个值，应该舍弃后面的一个值，使K减1，更新新的字符数组的下标值。直至将后面的一个值放在不大于前面的值的后面，
• 如果后面的值大于前面的一个值，直接放在新的字符数组中。

class Solution 
    public String removeKdigits(String num, int k) 
        //字符串判断
        if(num == null || num.length() == 0)
            return num;
        
        int length = num.length();
        if(k <= 0 || k > length)
            return num;
        
        if(k == length)
            return "0";
        

        //将字符串转为字符数组
        char[] chars = num.toCharArray();
        //保存移除k位数字后的数组 然后转为字符串
        char[] newchars = new char[length];

        int cur = 0;

        //从字符串的第一位开始判断
        for(int i = 0;i<length;i++)
            //判断移除位数是否大于0，然后比较结果，判断当前字符串的数字是否大于前一个数字，如果大于前一个数字舍弃。如果小于前一个数字
            //则更新新的字符数组，移除之前的字符
            while(k>0 && cur >0 && newchars[cur - 1] > chars[i])
                //移除一个数字
                k--;
                cur--;
            

            //将字符保存在新的字符串中
            newchars[cur] = chars[i];
            cur++;
        

        //直至将字符串遍历完
        //这是分为两种情况，前导0 和 K值判断
        
        //如果k大于0 说明遍历完字符串数组，后面的数字都是升序，需要移除后面的K个数
        if(k > 0)
            //更新保存返回字符串的数组下标
            cur = cur - k;
            k = 0;
        

        //前导0判断
        int startindex = 0;
        while(newchars[startindex] == '0' && startindex < cur )
            startindex++;
         

        //返回保存字符串数组的返回值，有起始值，[startindex,cur]
        
        if(cur - startindex > 0)
            StringBuilder sb = new StringBuilder();
            for(int i = startindex;i<cur;i++)
                sb.append(newchars[i]);
            
            return sb.toString();
        
        return "0";