LeetCode 80 删除有序数组中的重复项 II
给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
双指针
快慢指针(单向遍历双指针)
class Solution:def removeDuplicates(self, nums: List[int]) -> int:if not nums:return 0i, repeat_item, repeat_times = 1, nums[0], 1for idx in range(1, len(nums)):if nums[idx] == repeat_item:if repeat_times < 2:nums[i] = nums[idx]i += 1repeat_times += 1else:repeat_item = nums[idx]nums[i] = nums[idx]i += 1repeat_times = 1return i
看了看题解,发现常数可以从2开始,然后代码可以更加精简
class Solution(object):def removeDuplicates(self, nums):slow = 0for fast in range(len(nums)):if slow < 2 or nums[fast] != nums[slow - 2]:nums[slow] = nums[fast]slow += 1return slow# 作者:负雪明烛
# 链接:https://leetcode.cn/problems/remove-duplicates-from-sorted-array-ii/
# 来源:力扣(LeetCode)
# 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
同类题:算法练习7——买卖股票的最佳时机