Snapchat面试题:移除K位

描述
给定一个以字符串表示的非负整数,从该数字中移除掉k个数位,让剩余数位组成的数字尽可能小,求可能的最小结果。
数字的长度小于10002,并且大于等于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不为0,则删除栈顶元素。 最后将栈内的元素变为数字即可。

源代码

class Solution:
    """
    @param num: a string
    @param k: an integer
    @return: return a string
    """
    def removeKdigits(self, num, k):
        # write your code here
        
        
        
        if k == 0:
            return num
            
        if k >= len(num):
            return "0"
        
        result_list = []
        
        for i in range(len(num)):
            while len(result_list) > 0 and k > 0 and result_list[-1] > num[i]:
                result_list.pop()
                k -= 1 
            if num[i] != '0' or len(result_list) > 0 :
                result_list.append(num[i])
        
        while len(result_list) > 0 and k > 0:
            result_list.pop()
            k -= 1 
        
        if len(result_list) == 0:
            return '0'
        return ''.join(result_list)

更多题解参考

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值