LeetCode402. 移掉K位数字（单调栈）
作者：互联网
1、题目描述
https://leetcode-cn.com/problems/remove-k-digits/
给定一个以字符串表示的非负整数 num，移除这个数中的 k 位数字，使得剩下的数字最小。
输入: 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)
标签：digit,return,数字,丢弃,LeetCode402,num,移掉,stack,单调 来源： https://blog.csdn.net/IOT_victor/article/details/110493960