【LeetCode-数组】删除排序数组中的重复项
题目描述
给定一个排序数组,你需要在 原地 删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例:
给定 nums = [0,0,1,1,1,2,2,3,3,4],
函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4。
你不需要考虑数组中超出新长度后面的元素。
题目链接:https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array/
思路
做这题之前可以先做一下移除元素。使用一快一慢两个指针,i为慢指针,j为快指针,都从数组头开始遍历。如果nums[i]==nums[j],则j++,继续遍历下一个元素。如果nums[i]!=nums[j],则说明找到了不同的元素,因为重复的元素要保留一个,所以先将i+1后赋值,也就是nums[++i]=nums[j],当j遍历完毕时,当前数组的长度为i+1.代码如下:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if(nums.empty())
return 0;
int i=0, j=0;
int curLen = nums.size(); // 当前数组长度
while(j<curLen){
if(nums[i]!=nums[j]){
i++;
nums[i]=nums[j];
}
j++;
}
return i+1;
}
};
- 时间复杂度O(n)
j遍历一遍即可,所以O(n). - 空间复杂度O(1)
申请的额外内存不随输入规模的增大而增大。
相似题目
移除元素:https://leetcode-cn.com/problems/remove-element/
总结
当遇到移除元素的题目时,可以尝试使用一快一慢两个指针来解决。快指针慢指针起始位置相同,当快指针指向的元素与慢指针指向的元素不同时进行相应的操作。