402. 移掉K位数字

贪心的策略,每次删除一个数使得剩余数组成的数字最小。于是就从前往后找第一个i使得nums[i]>nums[j],删除该元素。重复该过程直至删除够k个元素,或者找不到这样的元素。但这种方法在最坏情况下的时间复杂度会达到O(n*k),如序列[1,2,3,...,100,99,98,..1,]

问题也可以从另一个角度来想:如果一个元素比它前面的元素小,我们就把它往前推一个(也就是删除前一个元素)直至它前面没有元素或者前面的元素比它小或者已经删除了k个。如果有多个满足条件的元素怎么办呢?由于两个相同位数的数字大小关系取决于第一个不同的数的大小,所以我们总是考虑第一个满足条件的元素。

再换一种想法就是用一个栈保存局部最优序列,每次用一个元素更新局部最优序列。

class Solution:
    def removeKdigits(self, num: str, k: int) -> str:
        if len(num)==k:
            return '0'

        n = len(num)
        res = []
        for i in range(n):
            while res and k>0 and num[i]<res[-1]:
                res.pop()
                k-=1
            res.append(num[i])
        
        if set(res) == {'0'}:
            return '0'
        elif k==0:
            return ''.join(res).lstrip('0') 
        else:
            return ''.join(res[:-k]).lstrip('0')
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值