leetcode-移掉k位数字-贪心算法(python)

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

思想:
从高位到低位遍历字符串,与后一个比较,大于则删除

代码:

class Solution:
    def removeKdigits(self, num, k):
        if k >= len(num): #k大于字符串长度,返回“0”
            return "0"
        i = 0
        while i < len(num) - 1 and k > 0:
            if int(num[i])>int(num[i+1]): #当前数大于后一个数
                num=num[0:i]+num[i+1:]    #删除当前数,前后拼接实现
                if i>0:
                    i-=1
                k-=1
            else:i+=1
        num=num[:len(num)-k]             #循环结束,k>0,当前字符串从小到大排列,则去掉最后k个数
        return str(int(num))             #int去掉前导0

猜你喜欢

转载自blog.csdn.net/weixin_40876685/article/details/88713397