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

问题

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

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

示例

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

解答

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int i = 0, k = 2; // 保留k个重复项
        for (int num : nums) {
            if (i < k || nums[i - k] != num)
                nums[i++] = num;
        }
        return i;
    }
};

重点思路

因为数组是有序的，当nums[i - k] != num时，必然能够保证当前num重复次数不超过k。