题目
给你一个有序数组 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] <= 104nums已按升序排列
思路
双指针做即可,和之前的去除重复项的思路都是一样的,抓住是有序数组,相同元素肯定会靠着即可。
代码
/**
* @param {number[]} nums
* @return {number}
*/
var removeDuplicates = function(nums) {
if (nums.length <= 2) {
return nums.length; // 不可能重复两个以上直接返回掉即可
}
// 因为是顺序的数组,所以重复项一定在一起,如果是两个以上的重复项,肯定下标-2还是这个元素
let slow = 2, fast = 2; // 双指针,从2开始就行
while (fast < nums.length) { //
if (nums[slow - 2] != nums[fast]) {
nums[slow] = nums[fast]; // 直接执行覆盖即可
++slow;
}
++fast;
}
return slow;
};
通解
这种去除有序数组重复项的问题,其实是具备双指针通用解法的:
我们可以发现:对于保留k位的相同数字的话,其实它的具体的方法是这样的:
- 对于前 k 个数字,我们可以直接保留
- 对于后面的任意数字,能够保留的前提是:与当前写入的位置前面的第 k 个元素进行比较,不相同则保留。 在这样的前提下,使用双指针形成下面的方法:
/**
* 保留有序数组的k位重复数字的通解
* @param {number[]} nums 有序数字数组
* @param {number} k 保留多少位
* @returns {number} 新数组的长度
*/
function removeDuplicates(nums, k){
if(nums.length <= k){ // 长度不允许你重复,直接返回即可
return nums.length;
}
let slow = k, fast = k; // 初始化快慢指针
while(fast < nums.length){
// 如果 nums[fast] 不等于 nums[slow - k]
// 则将 nums[fast] 复制到 nums[slow],并将 slow 向前移动一位
if (nums[fast] != nums[slow - k]) { //nums[slow - k] 是当前考虑的元素在新数组中的第一个可能的位置
nums[slow] = nums[fast];
++slow;
}
// 将 fast 向前移动一位,以检查下一个元素
++fast;
}
return slow; // 返回新数组长度
}
/**
* 展示数组用,因为返回的数组仅仅是依靠给定长度来实现展示的,真正的数组中含有无用元素
* @param {number[]} nums 数组
* @param {number} length 有效长度
*/
function displayArray(nums, length){
for(let i = 0; i<length; ++i){
console.log(nums[i]);
}
}
let nums = [1,1,1,1,1,2,2,2,3,3,4,5,6,7];
let len = removeDuplicates(nums, 1);
displayArray(nums, len); // 1234567
nums = [1,1,1,1,1,2,2,2,3,3,4,5,6,7];
len = removeDuplicates(nums, 2);
displayArray(nums, len); // 1122334567