一、题目
1、题目描述
给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例 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 。 不需要考虑数组中超出新长度后面的元素。
2、基础框架
3、原题链接
https://leetcode.cn/problems/remove-duplicates-from-sorted-array-ii/
二、解题报告
1、思路分析
(
1
)
(1)
(1)原地删除重复元素,这类题目的解题方法首先就要想到快慢指针。快指针负责遍历,慢指针负责修改数组。
(
2
)
(2)
(2)题目允许每个数字重复一次,所以我们设置一个计数器来记录每个数字重复次数,快慢指针初始时就错开。
(
3
)
(3)
(3)如果快慢指针所指的元素值相等,判断计数器值,如果计数器值等于2,说明当前元素的重复次数已经满了,不再需要了,需要将多余的重复数字删除,删除方法就是快指针一直遍历到和慢指针所指元素值不等的位置(不能越界),然后将当前快指针所指元素值放入慢指针前面的位置,同时快慢指针共同前进一步。计数器重新初始化。
(
4
)
(4)
(4)如果计数器值小于2,说明当前元素重复次数未满,将快指针所指元素放入慢指针前面一个位置,同时快慢指针前进一步,计数器值加1.
(
5
)
(5)
(5)如果快慢指针所指的元素值不相等,直接将快指针所指元素放入慢指针前面一个位置,同时快慢指针前进一步,计数器重新初始化
2、时间复杂度
最坏时间复杂度为O(n),空间复杂度为O(1)
3、代码详解
C++
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if (nums.size() < 3) return nums.size();
int f = 0;
int r = 1;
int cnt = 1;
while (r < nums.size()) {
if (nums[f] == nums[r]) {
if (cnt < 2) {
cnt++;
nums[++f] = nums[r++];
} else {
while (r < nums.size() && nums[f] == nums[r]) {
r++;
}
if (r < nums.size()) {
nums[++f] = nums[r++];
cnt = 1;
}
}
} else {
nums[++f] = nums[r++];
cnt = 1;
}
}
return f+1;
}
};
三、本题小知识
1、快慢指针的应用