描述
给定一个排序数组,在原数组中删除重复出现的数字,使得每个元素只出现一次,并且返回新的数组的长度。
不要使用额外的数组空间,必须在原地没有额外空间的条件下完成
样例
给出数组A =[1,1,2],你的函数应该返回长度2,此时A=[1,2]
考虑
- 排序数组, 说明重复出现的数字相邻
- 不用额外空间,在原有数组中进行操作
代码
// By Lentitude
class Solution {
public:
/**
* @param A: a list of integers
* @return : return an integer
*/
int removeDuplicates(vector<int> &nums) {
// write your code here
if (nums.empty()) return 0;
vector<int>::iterator pos = nums.begin();
for (int i = 0; i != nums.size(); ++i){
while (i + 1 < nums.size() && nums[i] == nums[i+1]){
nums.erase(pos + i + 1); // 如果相同, 则直接删除,动态数组会自动向前移动,直到前后数字不重复
}
}
return nums.size();
}
};