题目
给定一个排序数组,你需要在 原地 删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
给定 nums = [0,0,1,1,1,2,2,3,3,4],
函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4。
你不需要考虑数组中超出新长度后面的元素。
解答
方法一:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if (nums.empty() || nums.size() == 1) {
return nums.size();
}
int count = 1;
for(int i = 0; i < nums.size() - count;) {
///发现相等,把这个元素移到最后的位置,其他元素依次往前移动
if (nums[i] == nums[i + 1]) {
//把i+1元素移动到最后,
for (int j = i + 1; j < nums.size() - 1; j++) {
nums[j] = nums[j + 1];
}
//一次完整的移动之后,将count + 1
count++;
} else {
///不满足条件的时候才能移动下标
i++;
}
}
return nums.size() - count + 1;
}
};
方法二:
因为后边会有重复元素,我们只处理不重复的元素就行,只要不重复就往数组前边的位置放。
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if (nums.empty() || nums.size() == 1) {
return nums.size();
}
int count = 0;
for(int i = 0; i < nums.size() - 1; i++) {
//相等不处理,不相等放前边
if (nums[i] != nums[i + 1]) {
nums[count++] = nums[i];
}
}
nums[count] = nums.back();
return count + 1;
}
};