Snapchat面试题：移除K位

LintCode

于 2021-07-09 09:48:44 发布

阅读量208

点赞数

分类专栏： leetcode/lintcode每天一题文章标签： python 算法动态规划 leetcode

本文链接：https://blog.youkuaiyun.com/LintCode/article/details/118598458

版权

leetcode/lintcode每天一题专栏收录该内容

107 篇文章

订阅专栏

探讨一种算法，通过移除非负整数中的k个数位来构造最小可能的数字。采用单调栈实现，确保结果有效且无前导零。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

描述
给定一个以字符串表示的非负整数，从该数字中移除掉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)

更多题解参考