贪心算法——移除K位数字
给定一个以字符串表示的非负整数 num,移除这个数中的 k 位数字,使得剩下的数字最小。 注意: num 的长度小于 10002 且 ≥ k。 num 不会包含任何前导零。 例子: 输入: num = “1432219”, k = 3 输出: “1219” 解释: 移除掉三个数字 4, 3, 和 2 形成一个新的最小的数字 1219。
问题: 贪心策略 数据转换
解决: 贪心策略:有很多种移除一位的结果,但每种结果的位数都是相同的。相同位数的两个数判断大小的方法是什么? 较高位的大小 所以每次移除都要尽量使,结果的高位尽可能小。 所以从左到右比较,当高位大于次高位时,移除高位
数据转换: 因为涉及元素比较移除,所以字符串先转为列表 处理好后的列表在转为字符串 代码:
class Solution: def removeKdigits(self, num: str, k: int) -> str: nums=[i for i in num] for i in range(k): for j in range(len(nums)-1): if nums[j]>nums[j+1]: nums.pop(j) break else: nums.pop() j=0 #因为类似【’1‘,‘0’,’2‘】,k=1 结果为【‘0’,’2‘】,所以列表最后要去除以0开头的连续0元素,记录第一个非0元素位置 for i in nums: if i==0: j+=1 else: break #因为类似【‘0’,‘0’】 会在上一步变为【】,所以要判断 return .join(nums[j:]) if j!=len(nums) else 0