【leetcode刷题】T176-移掉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。

class Solution(object):
    def removeKdigits(self, num, k):
        """
        :type num: str
        :type k: int
        :rtype: str
        """
        if k >= len(num):
            return '0'
        stack = []
        for i, n in enumerate(num):
            if k == 0:
                stack.append(num[i:])
                break
            if len(stack) == 0 or n >= stack[-1]:
                stack.append(n)
            else:
                while len(stack) > 0 and n < stack[-1]:
                    stack.pop(-1)
                    k -= 1
                    if k == 0:
                        break
                stack.append(num[i])

        while k != 0:
            stack.pop(-1)
            k -= 1
        res = ''.join(stack).lstrip('0')
        if res == '':
            res = '0'
        return res