【11月打卡~Leetcode每日一题】402. 移掉K位数字(难度:中等)

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

注意:

num 的长度小于 10002 且 ≥ k。
num 不会包含任何前导零。

贪心的思路,每次从开头循环,如果当前字符大于之后的一个字符,则删除,即可满足题意,但是速度比较慢,需要优化

class Solution:
    def removeKdigits(self, num: str, k: int) -> str:
        while(k):
            k -= 1
            flag = True
            for i in range(len(num)-1):
                if flag and num[i]>num[i+1]:
                    num = num[:i]+num[i+1:]
                    flag = False
            if flag:
                num = num[:-1]
                
        num = num.lstrip("0")
        return num if num else "0"

加个位置标识符 index,可以大大减少迭代时间
比如 num = “1234567890”
如果按照上边的代码,每次都会从第0位开始循环,但是如果每次只去除一个字符的话,只需要从删除的字符上一个位置开始搜索即可
这样可以把理论上的时间复杂度从O(nk)变成O(n),其中k表示删除的字符数,n表示字符串长度,只需注意如果删除的是第0个字符时,要将index赋值成0,而不是-1

class Solution:
    def removeKdigits(self, num: str, k: int) -> str:
        index = 0
        while(k):
            k -= 1
            flag = True
            for i in range(index,len(num)-1):
                if num[i]>num[i+1]:
                    num = num[:i]+num[i+1:]
                    flag = False
                    index = i-1 if i-1>=0 else 0
                    break
            if flag:
                num = num[:-1]
                
        num = num.lstrip("0")
        return num if num else "0"
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值