链接
https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array/
题意
给定一个 排序数组,在 原地 删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,在使用 O(1) 额外空间的条件下完成。
思路
设置两个指针,j 指向目前为止不重复的最后一个元素,i 为移动指针,向后找不重复的元素。也就是说当nums[j] != nums[i]
时,找到了不重复的元素,将 j 指向的后一个元素替换成nums[i]
,直到 i 指到最后一个元素。
PS:
- 因为 j 是指向不重复的最后一个元素的位置,所以不重复的元素个数就是 j+1。
- 注意数组为空时的情况,在这种做法下,数组为空也会返回 ,造成溢出,需要特判。
AC代码
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if(nums.empty()) return 0; // 在这种做法下,数组为空也会返回 1,造成溢出,需要特判
int n = nums.size();
int j = 0;
for(int i = 1; i < n; ++i) {
if(nums[j] != nums[i]) {
j++;
nums[j] = nums[i];
}
}
return j+1; // j 是不重复的最后一个元素的位置,不重复的元素个数 = j+1
}
};