题目描述
给定一个排序数组,你需要在 原地 删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
思路
- 用count记录非重复数字,count_dup记录重复数字,从nums[1]开始遍历,若nums[i] == nums[i-1],则count_dup++,并将nums[i]删掉,i不变;若不同,则coount++。最终当i==size-count_dup时结束
代码
方法一:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if(nums.size() < 2)
return nums.size();
int count = 1;
int count_dup = 0;
int size = nums.size();
for (int i = 1; i < size-count_dup;)
{
if (nums[i] == nums[i - 1])
{
count_dup++;
nums.erase(nums.begin() + i);
}
else
{
count++;
i++;
}
}
return count;
}
};