删除排序数组中的重复数字

描述

给定一个排序数组，在原数组中删除重复出现的数字，使得每个元素只出现一次，并且返回新的数组的长度。

不要使用额外的数组空间，必须在原地没有额外空间的条件下完成

样例

给出数组A =[1,1,2]，你的函数应该返回长度2，此时A=[1,2]

考虑

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();

     }
};