Given a sorted array, remove the duplicates in-place such that each element appear only once and return the new length.
Do not allocate extra space for another array, you must do this by modifying the input array in-place with O(1) extra memory.
Example:
Given nums = [1,1,2],
Your function should return length = 2, with the first two elements of nums being 1 and 2 respectively.
It doesn't matter what you leave beyond the new length.
注意:数组是有序的
定义一个区间[0…k),里面存放的元素唯一。
初始k=1,表示区间[0…1)内存在一个不重复的元素,也就是数组的第一个元素。
索引k表示下一个不重复的元素将要放置的位置,同时也表示不重复元素的个数。当数组的size为1时,直接返回k。
大体思路:从索引为1的元素开始遍历。当nums[i] == nums[i-1]时,说明i位置的元素是重复的,继续向下遍历。如果nums[i] != nums[i-1]时,说明该元素是第一次出现(从前向后),那么将它放置到k位置,同时k++。
优化:如果k == i,说明当前元素已经放置到我们前面的定义要求的位置,那么只需要k++。
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if (nums.size() == 0) return 0;
int k = 1;
for (int i = 1; i < nums.size(); i++) {
if (nums[i] != nums[i - 1]) {
if (k != i)
nums[k++] = nums[i];
else
k++;
}
}
return k;
}
};