一、题目描述:
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii
二、解题思路
思路一:
双指针法,一次遍历数组,用一个变量n来保存当前指针数字的相同个数,n初始值为1;若相邻两个数相等,则n = n+1,不相等时将n置1,;若 n > 2了,则说明至少有3个重复的值了,用splice删除。
时间复杂度:O(n)
空间复杂度:O(1)
var removeDuplicates = function (nums) {
let len = nums.length;
let n = 1;
for (let i = 0; i < len - 1; i++) {
if (nums[i] == nums[i + 1]) {
n++;
} else {
n = 1;
}
if (n > 2) {
nums.splice(i + 1, 1);
i--;
}
}
return nums.length;
};
思路二:
在思路一的基础上进行改进,题目中说道,每个元素最多出现两次,那就没有必要比较相邻的数了,只需要隔一个数比较,即第i 个和第 i+2 个,相同则删除,不同则向后移指针。
时间复杂度:O(n)
空间复杂度:O(1)
var removeDuplicates = function (nums) {
let len = nums.length;
let i = 0;
while (i < len - 2) {
if (nums[i] == nums[i + 2]) {
nums.splice(i, 1);
} else {
i++;
}
}
return nums.length;
};