华为OD-移掉K位数字

题目描述

给定一个以字符串表示的数字 num 和一个数字 k ,从 num 中移除 k 位数字,使得剩下的数字最小。如果可以删除全部数字,则结果为 0。

1.num仅有数字组成

2.num是合法的数字,不含前导0

3.删除之后的num,请去掉前导0(不算在移除次数中)

数据范围:num的长度满足 1≤105 1≤k≤n≤105  ,保证 num 中仅包含 0~9 的十进制数

输入描述

第一行输入一个字符串数字,一个数字K

输出描述

剩余的数字

示例一

输入

"1432219",3

输出

"1219"
 

示例二

输入

"100999",3

输出

"9"

代码实现

# coding:utf-8

# 移掉 K 位数字
# https://www.nowcoder.com/practice/0fe685c8272d40f1b9785fedd2499c1c?tpId=196&tqId=39412&ru=/exam/oj
"""
思路:
        核心:以某字符为中心,寻找左边的最近比该字符严格小的字符。(单调栈:严格单调递增)
        步骤:
            从左到右遍历字符串,如果栈非空&&栈顶元素>=当前字符&&k!=0,则弹栈,并最后将当前字符入栈。
            当k==0时,则说明弹栈次数用完了。
            当弹出0的时候,不需k--,因为前导0忽略。
            最后,当弹栈次数用完后,只需将栈中字符和未入栈字符拼接,即可得到数值最小化。
"""
class Solution:
    def removeKnums(self, num, k):
        # write code here
        stack = []
        for i in range(len(num)):
            while stack and k and int(num[i]) < int(stack[-1]):
                k -= 1
                stack.pop()
            if not stack and num[i] == '0':
                continue
            stack.append(num[i])
        stack = stack[:-k] if k else stack
        if not stack:
            stack.append('0')
        return ''.join(stack)

猜你喜欢

转载自blog.csdn.net/SD_JZZ/article/details/132404587
今日推荐