ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

LeetCode402. 移掉K位数字(单调栈)

2020-12-02 18:31:31  阅读:249  来源: 互联网

标签:digit return 数字 丢弃 LeetCode402 num 移掉 stack 单调


1、题目描述

https://leetcode-cn.com/problems/remove-k-digits/

给定一个以字符串表示的非负整数 num,移除这个数中的 位数字,使得剩下的数字最小。

输入: num = "1432219", k = 3
输出: "1219"
解释: 移除掉三个数字 4, 3, 和 2 形成一个新的最小的数字 1219。
输入: num = "10200", k = 1
输出: "200"
解释: 移掉首位的 1 剩下的数字为 200. 注意输出不能有任何前导零。
​输入: num = "10", k = 2
输出: "0"
解释: 从原数字移除所有的数字,剩余为空就是0。

 

2、代码详解

使用单调栈

由于 4 比左侧相邻的 1 大。如果选择丢弃左侧的 1,那么会使得剩下的数字更大(开头的数从 1 变成了 4)。因此选择不丢弃。

由于 3 比左侧相邻的 4 小。 如果选择丢弃左侧的 4,那么会使得剩下的数字更小(开头的数从 4 变成了 3)。因此选择丢弃

代码思路:

  • 遇到小于栈顶的数,丢弃
  • 保留 n - k 个元素。其中 n 为数字长度。 只需要遍历完成之后,再截取前n - k个元素即可。  
  • 按由于我们需要保留和丢弃相邻的元素,因此使用栈这种在一端进行添加和删除。
class Solution(object):
    def removeKdigits(self, num, k):
        """
        :type num: str
        :type k: int
        :rtype: str
        """
        stack = []
        remain = len(num) - k
        for digit in num:
            while k and stack and stack[-1] > digit:
                stack.pop()  # 遇到小于栈顶的数,丢弃
                k -= 1
            stack.append(digit)
        # return ''.join(stack[:remain]).lstrip('0') or '0'
        if len(stack) > 0:
            s = ''.join(stack[:remain]).lstrip('0')
            if s == '':
                return '0'
            else:
                return s
        else:
            return '0'

# 特例
num = "10"
k = 2
s = Solution()
print(s.removeKdigits(num, k))

O(N) O(N)

https://leetcode-cn.com/problems/remove-k-digits/solution/yi-zhao-chi-bian-li-kou-si-dao-ti-ma-ma-zai-ye-b-5/

标签:digit,return,数字,丢弃,LeetCode402,num,移掉,stack,单调
来源: https://blog.csdn.net/IOT_victor/article/details/110493960

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有