算法练习4——删除有序数组中的重复项 II

LeetCode 80 删除有序数组中的重复项 II

给你一个有序数组 nums ，请你 原地 删除重复出现的元素，使得出现次数超过两次的元素只出现两次 ，返回删除后数组的新长度。
不要使用额外的数组空间，你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。

双指针
快慢指针（单向遍历双指针）

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        if not nums:
            return 0
        i, repeat_item, repeat_times = 1, nums[0], 1
        for idx in range(1, len(nums)):
            if nums[idx] == repeat_item:
                if repeat_times < 2:
                    nums[i] = nums[idx]
                    i += 1
                repeat_times += 1
            else:
                repeat_item = nums[idx]
                nums[i] = nums[idx]
                i += 1
                repeat_times = 1
        return i
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

看了看题解，发现常数可以从2开始，然后代码可以更加精简

class Solution(object):
    def removeDuplicates(self, nums):
        slow = 0
        for fast in range(len(nums)):
            if slow < 2 or nums[fast] != nums[slow - 2]:
                nums[slow] = nums[fast]
                slow += 1
        return slow

# 作者：负雪明烛
# 链接：https://leetcode.cn/problems/remove-duplicates-from-sorted-array-ii/
# 来源：力扣（LeetCode）
# 著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。
1
2
3
4
5
6
7
8
9
10
11
12
13

同类题：算法练习7——买卖股票的最佳时机