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

原题链接： 80. 删除有序数组中的重复项 II

题目大意

给你一个有序数组 nums ，请你 原地 删除重复出现的元素，使每个元素 最多出现两次 ，返回删除后数组的新长度。

不要使用额外的数组空间，你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例：

输入：nums = [1,1,1,2,2,3]
输出：5, nums = [1,1,2,2,3]

输入：nums = [0,0,1,1,1,1,2,3,3]
输出：7, nums = [0,0,1,1,2,3,3]

数据范围：

1 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
nums 已按升序排列

思路

维护一个最终返回的数组，长度为k。
如果当前长度小于2那肯定新的数有效，就加入维护的数组(nums[k ++]=x)；
长度达到2之后，新的数只要不跟维护的数组的后两个都相同，则有效，加进去。否则，就跳过这个新的数。
最终k就是维护数组的长度。
一些精妙之处：
原数组是有序的，而维护的答案数组也是有序的。所以维护的这个数组重复数≤2只需要加进来的数x都满足x!=nums[left-2]即可。

源代码

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int k = 0;
        for(auto x : nums){
            if(k < 2 || nums[k - 2] != x){
                nums[k ++ ] = x;
            }
        }

        return k;
    }
};

复杂度分析

时间复杂度：O(n)
空间复杂度：O(1) 原地算法

类似题目

看完本文，你可以解决如下题目：
26. 删除有序数组中的重复项
83. 删除排序链表中的重复元素
82. 删除排序链表中的重复元素 II