JAVA练习362- 删除有序数组中的重复项 II

题目概览

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

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

说明：

为什么返回数值是整数，但输出的答案是数组呢？

请注意，输入数组是以「引用」方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下:

//nums是以“引用”方式传递的。也就是说，不对实参做任何拷贝 int len = removeDuplicates(nums); // 在函数里修改输入数组对于调用者是可见的。 // 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。 for (int i = 0; i < len; i++) { print(nums[i]); }

示例 1：

输入：nums = [1,1,1,2,2,3]输出：5, nums = [1,1,2,2,3]解释：函数应返回新长度 length =5, 并且原数组的前五个元素被修改为1, 1, 2, 2, 3。 不需要考虑数组中超出新长度后面的元素。

示例 2：

输入：nums = [0,0,1,1,1,1,2,3,3]输出：7, nums = [0,0,1,1,2,3,3]解释：函数应返回新长度 length =7, 并且原数组的前七个元素被修改为0, 0, 1, 1, 2, 3, 3。不需要考虑数组中超出新长度后面的元素。

提示：

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

来源：80. 删除有序数组中的重复项 II - 力扣（LeetCode）

解题分析

方法：三指针

定义三个指针 i，j，k 分别指向 头、头+1、头，那么：

1. 当 nums[ i ] == nums[ j ] 时，有重复元素，不断移动 j 即 j++ 直到不相等，然后将 nums[ k ] 设置为 nums[ i ]，由于我们是要保留两个元素，因此还要看 j - i 是否大于 1，大于 1 继续将 nums[ k + 1 ] 设置为 nums[ i ]，最后 i 移动到 j 的位置，j 往后移动一位，k 移动到最近一个没设置值的位置，继续重复操作
2. 当 nums[ i ] != nums[ j ] 时，没有重复元素，将 nums[ k ] 设置为 nums[ i ]，所有指针往后移动一位即可
3. 当 k >= n 时，结束遍历

时间复杂度：O(n)
空间复杂度：O(1)

class Solution { public int removeDuplicates(int[] nums) { int n = nums.length; if (n == 0) { return 0; } int i = 0, j = 1, k = 0; while(i < n && k < n) { while(j < n && nums[i] == nums[j]) { j++; } nums[k++] = nums[i]; if (j - i > 1) { nums[k++] = nums[i]; } i = j; j++; } return k; } }