Problem
给定一个排序数组,你需要在 原地 删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例:
输入 | 输出 |
---|---|
[1,1,2] | 2 (并且原数组 nums 的前2个元素被修改为 1, 2。不需要考虑数组中超出新长度后面的元素。) |
[0,0,1,1,1,2,2,3,3,4] | 5 (并且原数组 nums 的前5个元素被修改为 0, 1, 2, 3, 4。不需要考虑数组中超出新长度后面的元素。) |
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以 “引用” 方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参做任何拷贝
int len = removeDuplicates(nums);
// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。
for (int i = 0; i < len; i++) {
print(nums[i]);
}
Solving
题目要求原地删除数组中重复的元素,本人的想法是使用 erase() 函数直接将重复出现的元素从数组中删除掉(以下代码中的直接删除法),但此方法等同于直接将原数组的长度改变了,仔细思考其实与题目表达的原地删除并不符,因此可以采用双指针法来实现。
双指针法是用两个指针 i 和 j 依次进行索引,其中指针 i 用于索引每一类重复元素的第一个,指针 j 则依次后移直至指针 i 与 j 指向的元素不同为止,此时如果 i 与 j 不相邻,则将将指针 j 指向的元素复制给指针 i 指向的下一个元素(即第 i+1 个元素),同时 i 和 j 后移;如果 i 与 j 相邻,则不需要复制元素,直接将 i 和 j 后移。
Code(C++)
直接删除法:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int i = 0;
while(i < nums.size()-1 && nums.size() != 0){
if(nums[i] == nums[i+1]){
nums.erase(nums.begin() + i + 1);
}
else{
i++;
}
}
return nums.size();
}
};
执行用时:288 ms
内存消耗:10 MB
双指针法:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if(nums.size() == 0 || nums.size() == 1) //当数组为空或者只有一个元素时直接返回数组的大小;
return nums.size();
int i = 0;
for(int j = 1; j < nums.size(); j++){
if(nums[i] != nums[j]){
if(j-i > 1){ //当指针i和j不相邻指向的元素不同时,将指针j指向的元素复制给指针i指向的下一个元素;
nums[i+1] = nums[j];
}
i++;
}
}
return i+1;
}
};
执行用时:16 ms
内存消耗:9.2 MB