LeetCode80. 删除有序数组中的重复项 II题解
题目链接:
https://leetcode.cn/problems/remove-duplicates-from-sorted-array-ii/
题目描述:
给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例:
输入:nums = [0,0,1,1,1,1,2,3,3,3]
输出:7, nums = [0,0,1,1,2,3,3]
解释:函数应返回新长度 length = 7, 并且原数组的前七个元素被修改为 0, 0, 1, 1, 2, 3, 3。不需要考虑数组中超出新长度后面的元素。
思路
以题目给出的示例而言,我们可以把数组分为两个部分:
前一部分是不需要删除的,也就是每个数字重复数量不大于2的;
后一部分是需要删除的;
我们可以定义快慢指针:
我们可以定义快慢指针:
[0,slow-1]代表不需要删除区域;
[slow,fast-1]代表需要重复的区域;
[fast,n-1]代表未处理的区域。
循环逻辑
- 初始化快慢指针slow,fast,均初始化为2(注意:前两个元素肯定重复次数不大于2);
- 如果nums[fast] 不等于 nums[slow - 2],说明遇到重复次数不超过2的元素,那么就把 nums[slow] 赋值为nums[fast],并把slow指针向后移动;
- fast指针向后移动
直到fast超出数组长度。
然后返回slow的值,就是所求数组长度。
代码如下:
class Solution {
public int removeDuplicates(int[] nums) {
if(nums == null || nums.length == 0) return 0;
if(nums.length <= 2) return nums.length;
int slow = 2,fast =2;
while(fast < nums.length){
if(nums[fast] != nums[slow - 2]){
nums[slow] = nums[fast];
slow++;
}
fast++;
}
return slow;
}
}