原题链接: 80. 删除有序数组中的重复项 II
题目大意
给你一个有序数组 nums
,请你 原地 删除重复出现的元素,使每个元素 最多出现两次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例:
输入:nums = [1,1,1,2,2,3]
输出:5, nums = [1,1,2,2,3]
输入:nums = [0,0,1,1,1,1,2,3,3]
输出:7, nums = [0,0,1,1,2,3,3]
数据范围:
1 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
nums 已按升序排列
思路
维护一个最终返回的数组,长度为k。
如果当前长度小于2那肯定新的数有效,就加入维护的数组(nums[k ++]=x);
长度达到2之后,新的数只要不跟维护的数组的后两个都相同,则有效,加进去。否则,就跳过这个新的数。
最终k就是维护数组的长度。
一些精妙之处:
原数组是有序的,而维护的答案数组也是有序的。所以维护的这个数组重复数≤2只需要加进来的数x都满足x!=nums[left-2]即可。
源代码
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int k = 0;
for(auto x : nums){
if(k < 2 || nums[k - 2] != x){
nums[k ++ ] = x;
}
}
return k;
}
};
复杂度分析
时间复杂度:O(n)
空间复杂度:O(1) 原地算法
类似题目
看完本文,你可以解决如下题目:
26. 删除有序数组中的重复项
83. 删除排序链表中的重复元素
82. 删除排序链表中的重复元素 II