题目来源。
给你一个有序数组 nums
,请你** 原地** 删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
思路1:移动元素法。从后遍历数组,计数元素重复次数,当遇到不重复元素且重复数量大于2时,开始把后面子数组往前移,缩小数组长度。
/*** @param {number[]} nums* @return {number}*/
var compressArry = function (nums, i, j) {let n = nums.length;for (let k = i; k < n; k++) {nums[k - j] = nums[k];}nums.length -= j;
}
var removeDuplicates = function (nums) {let i = nums.length - 1,j = 1;while (i - j >= 0) {if (nums[i - j] !== nums[i]) {//这时候的j就是重复次数if (j > 2) {//将[i,nums.length-1]往前移j-2位compressArry(nums, i, j - 2);}//不管大于2还是小于等于2,i在这时都要移动到i-j,开始下一轮计数i = i - j;j = 1;} else {j++;}}//最前面的重复数还没有清理,现在开始清理if (j > 2) {//将[i,nums.length-1]往前移j-2位compressArry(nums, i, j - 2);}return nums.length;
};
时间复杂度: O ( n 2 ) O(n^2) O(n2),
空间复杂度: O ( 1 ) O(1) O(1)。
思路2:重复次数达到3,立马删除。
/*** @param {number[]} nums* @return {number}*/
var removeDuplicates = function(nums) {if(nums.length<=2) return nums.length;let i=0,j=2;while(i<nums.length){if(nums[i]===nums[j]){nums.splice(j,1);}else{i++;j++;}}return nums.length;
}
时间复杂度:O(n2)
空间复杂度:O(1)
思路3:双指针法。
/*** @param {number[]} nums* @return {number}*/
var removeDuplicates = function (nums) {if(nums.length<=2){return nums.length;}let slow=2;for(let fast=2;fast<nums.length;fast++){if(nums[fast]!==nums[slow-2]){//找到新元素nums[slow++]=nums[fast];}}nums.length=slow;return slow;
};
解释:slow指针保证了其前面的元素重复次数不超过2,fast指针则是向后寻找新元素,一旦找到新元素,将其插入到slow的位置。接着,slow指针往后移一位,fast继续寻找。如果fast找到数组末尾了,那么说明新元素寻找完毕,故设置数组长度为slow即为满足需要的数组。
时间复杂度:O(n)
空间复杂度:O(1)