【leetcode】80 删除排序数组中的重复项 II（数组，双指针）

题目链接：https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii/

题目描述

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素最多出现两次，返回移除后数组的新长度。

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

示例 1:

给定 nums = [1,1,1,2,2,3],

函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。

你不需要考虑数组中超出新长度后面的元素。

示例 2:

给定 nums = [0,0,1,1,1,1,2,3,3],

函数应返回新长度 length = 7, 并且原数组的前五个元素被修改为 0, 0, 1, 1, 2, 3, 3 。

你不需要考虑数组中超出新长度后面的元素。

思路

• 原地删除肯定是双指针，一个指向遍历的元素，一个指向可以写入的位置，后者的大小是小于等于前者的，关键在于题目条件的转化，如何实现限制最多两次的重复出现。
• 我们先不考虑边界情况，只考虑中间的情况，假设当前遍历位置为i，写指针的可写入位置为current+1，对于i处的值，其写入的条件是重复小于等于2次，我们考虑已经写入的最后两位current和current-1，这两个位置的情况有两个，相等和不相等，首先考虑相等的情况，此时若i处的值和current-1或者说current处的值相同，那么，i处的值肯定不能加入；然后考虑不相等的情况，即current-1和current处值不相等，那么i处的值无论为什么，都满足题意的，即可以加入。
• 综上所述，当i处的值与current-1处的值不相等时，i处的值可以加入，其他情况均不能加入。
• 接着考虑边界情况，我们只需要考虑开始即可，开始时，前两个值无论等还是不等，都要原封不动的挪到新数组里，由于新数组就是在原数组上进行修改的，因此前两位直接不动即可，只需要修改遍历指针和写入指针就行。

复杂度分析
以上算法只需要进行一次遍历即可，时间复杂度O(n)，空间复杂度O(1)。

代码

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if(nums.size() <= 2) 
            return nums.size();
        int curr = 1;   // 新数组中有效位置的最后一位，新加入的数据应当写到curr+1
        for (int i = 2; i < nums.size(); ++i) { // 前2个必定符合，从第三个开始遍历
            if(nums[i] != nums[curr-1]){
                curr ++;
                nums[curr] = nums[i];
            }
        }
        return curr + 1;
    }
};

上面的思路同样地可用于每个元素最多出现k次：只要把curr-1的判断部分改为curr-k+1即可！