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