LeetCode 402. 移掉K位数字

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。

解题思路：
首先，我们使用类似于贪心的方式，从最高位起，选择尽可能小的数字，所以，当我们发现前一位大于后一位的时候，我们将其删掉并且更新K值（因为最后剩下的数字的相对位置一定与现在一样），当我们按照这样的方式遍历之后，如果K值还没有用完，此时的数字序列是从小到大排列的，所以我们删掉后面的K位。最后，将首位的0全部删除，得到结果

代码如下：

class Solution:
    def removeKdigits(self, num, k):
        """
        :type num: str
        :type k: int
        :rtype: str
        """
        if len(num) ==0 or len(num) == k:
            return '0'
        if k == 0:
            return num
        i=0
        while i<len(num)-1 and k>0:
            if num[i]>num[i+1]:
                num=num[:i]+num[i+1:]
                k=k-1
                if i>0:
                    i=i-1
            else:
                i=i+1
        if k>0:
            num=num[:len(num)-k]
        while len(num)>1 and num[0]=='0':
            num=num[1:]
        return num