问题描述
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.
思路分析
给出一个已排序的数组,要求去除掉重复的部分,使得每个元素只出现一次,并返回新数组的长度。in-place是指不引入新的数据结构,只对输入进行操作完成算法目的。思路是使用两个indexes,一个用于遍历数组(i index),另一个用于在有不同值出现的时候,将其插入到正确的位置并记录数组长度(count index)。
代码
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if (nums.size() <= 1)
return nums.size();
int count = 1;
for(int i = 1; i < nums.size(); i++){
if (nums[i] != nums[i-1]){
nums[count] = nums[i];
count++;
}
}
return count;
}
};
反思
一开始理解错了题目要求,只count计算了非重复值的元素个数,即新的长度,没有对原数组进行操作,所以各种出错。