//给你一个有序数组 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 已按升序排列
//
// Related Topics 数组 双指针
// 👍 432 👎 0
移除超过两个相同的项。
因为题目给出的数组是有序的,所以必然相同的元素是相邻的。
双指针
因此可以使用双指针将相同的元素框出来,再移动到数组前列。
如果框出来的元素超过或等于两个,就只移动两个过去,如果不足两个,就只移动一个。
class Solution {
public int removeDuplicates(int[] nums) {
int n = nums.length;
// 标记当前数组前面已经放了多少个元素
int index = 0;
int i = 0;
while (i < n) {
// 使用[left, right)去框选相同的元素
int left = i;
int right = i + 1;
// 如果left == right,right就右移
while (right < n && nums[right] == nums[left]) {
right ++;
}
// 如果相同的元素数量大于等于2,只移动两个过去
if (right - left > 1) {
// 有需要删除的数据
nums[index] = nums[left];
nums[index + 1] = nums[left];
index += 2;
}else {
// 否则说明只有一个元素,只移动一个过去
nums[index] = nums[left];
index += 1;
}
// 此时i已经判断完[left, right - 1]
i = right;
}
return index;
}
}
相同元素计数
因为只要求重复,所以我们只对前两个或者不重复的数字感兴趣,对于什么时候结束重复兴趣不大。
所以可以只记录当前i指示的数字重复了多少次,当小于等于2的时候才移动到数组头。大于2的时候就直接不感兴趣了。
class Solution {
public int removeDuplicates(int[] nums) {
int i = 1;
int count = 1;
for (int j = 1; j < nums.length; j++) {
// 如果当前数和前面相等的话,计数+1
if (nums[j] == nums[j - 1]) {
count ++;
}else {
// 如果不相等,说明换成了新的数字,计数回到1
count = 1;
}
// 如果计数小于等于2,这时移动到数组前部,如果超过2就直接抛弃
if (count <= 2) {
nums[i] = nums[j];
i ++;
}
}
return i;
}
}
直接移动
如果数组i位置的数相同超过2个,那么到达i的时候 nums[i - 2] == nums[i]
因此可以使用这个特点,不再计数,直接判断是否当前数字已经满了。
class Solution {
public int removeDuplicates(int[] nums) {
int i = 0;
for (int num : nums) {
// i < 2是因为,无论数组长什么样,前两个数字肯定是符合要求的
// i指示的是当前应该放数字的位置。 如果向前两个位置和当前数字相同,意味着相同数字已经足够了,可以直接抛弃
if (i < 2 || nums[i - 2] < num) {
nums[i] = num;
i ++;
}
}
return i;
}
}