leetcode----80. Remove Duplicates from Sorted Array II

链接：

https://leetcode.com/problems/remove-duplicates-from-sorted-array-ii/

大意：

给一个有序数组nums，nums中可能有些值出现多次。现在需要对nums做一个操作，使得nums中任何元素出现次数最多为2（即原来出现次数为1的元素在“新数组”中出现次数仍然为1，出现次数为3的元素在“新数组”中出现次数为2），并返回“新数组”的长度。要求：不能使用额外存储空间。例子：

思路：

记录一个变量count用于统计当前元素出现的次数，变量curIdx为在“新数组”中元素应该放入的位置，i为当前数组遍历到的位置。

情况一：如果当前元素和前一元素相同

• 当count<=2时，进行操作：nums[curIdx++] = nums[i++]
• 当count > 2时，只进行i++

情况二：如果当前元素和前一元素不同

则进行操作：nums[curIdx++] = nums[i++]； 并且 count重置为1

代码：

class Solution {
    public int removeDuplicates(int[] nums) {
        if (nums.length <= 2)
            return nums.length;
        int count = 1, curIdx = 1, i = 1;
        while (i < nums.length) {
            if (nums[i] == nums[i - 1]) {
                count++;
                if (count <= 2) {
                    nums[curIdx++] = nums[i];
                }
            } else {
                nums[curIdx++] = nums[i];
                count = 1;
            }
            i++;
        }
        return curIdx;
    }
}

结果：

结论：

逻辑