码迷,mamicode.com
首页 > 编程语言 > 详细

算法46----移除K位数字

时间:2018-11-11 20:18:26      阅读:227      评论:0      收藏:0      [点我收藏+]

标签:整数   not   print   元素   ==   ret   --   pre   move   

一、题目:移除K位数字

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

注意:

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

示例 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。

思路1:

采用一个栈:若栈最后一个元素比num中当前元素大,则存入栈中,否则将栈中最后一个元素删除。当K==0时停止。

代码1:

    def removeKdigits(self, num, k):
        """
        :type num: str
        :type k: int
        :rtype: str
        """
        #采用栈
        if not num:
            return None
        if k == 0:
            return num
        
        stack =[]
        res = ""

        for n in num:
            while k>0 and stack and int(stack[-1]) > int(n):
                k -= 1
                stack.pop()
            if n.isdigit():
                stack.append(n)
        print(stack)
        res = "".join(stack)
        if k>0:
            m = len(stack)-k
            res = "".join(stack[:m])     
        print(res)
        return str(int(res)) if res else "0"

 

思路2:

若num前一个元素比后一个元素大,则删除。直到k==0

代码2:

    def removeKdigits(self, num, k):
        if k>len(num)-1:
            return "0"
        i = 0
        while i< len(num)-1 and k>0:
            if int(num[i]) > int(num[i+1]):
                num = num[:i]+num[i+1:]
                if i>0:
                    i-=1
                k-=1
            else:
                i+=1
        num = num[:len(num)-k]
        return str(int(num))

 

算法46----移除K位数字

标签:整数   not   print   元素   ==   ret   --   pre   move   

原文地址:https://www.cnblogs.com/Lee-yl/p/9942990.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!