题目
给你一个 升序排列 的数组nums
,请你原地删除重复出现的元素,使每个元素只出现一次,返回删除后数组的新长度。元素的相对顺序应该保持 一致 。
由于在某些语言中不能改变数组的长度,所以必须将结果放在数组nums的第一部分。更规范地说,如果在删除重复项之后有k
个元素,那么nums
的前k
个元素应该保存最终结果。
将最终结果插入nums
的前k
个位置后返回k
。
不要使用额外的空间,你必须在原地修改输入数组并在使用O(1)
额外空间的条件下完成。
判题标准:
系统会用下面的代码来测试你的题解:
int[] nums = [...]; // 输入数组
int[] expectedNums = [...]; // 长度正确的期望答案
int k = removeDuplicates(nums); // 调用
assert k == expectedNums.length;
for (int i = 0; i < k; i++) {
assert nums[i] == expectedNums[i];
}
如果所有断言都通过,那么您的题解将被 通过。
示例 1:
输入:nums = [1,1,2]
输出:2, nums = [1,2,_]
解释:函数应该返回新的长度 2 ,并且原数组 nums 的前两个元素被修改为 1, 2 。不需要考虑数组中超出
新长度后面的元素。
提示:
0 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
nums 已按升序排列
解决方案
双指针法
采用快慢指针,快指针遍历寻找不同元素,慢指针更改数组。
循环条件:
- 快慢指针分别从下标
1,0
处开始 - 快指针遍历整个数组后停止
指针移动条件:
- 快指针每次向后移动一位
- 当两指针指向元素不同时,慢指针向后移动一位,并赋值为快指针指向元素
该方法时间复杂度为O(n)
。
代码:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if(nums.size() < 2) return nums.size();
int left = 0;
int right = left+1;
while(right<nums.size()) {
if(nums[left] != nums[right]) {
nums[++left] = nums[right];
}
right++;
}
return left+1;
}
};