一、题目:移除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))
原文地址:https://www.cnblogs.com/Lee-yl/p/9942990.html
时间: 2024-10-29 16:27:19