删除排序数组中的重复项-LeetCode

题目来源:

https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array/description/

---

描述:

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素只出现一次，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

---

示例：

给定 nums = [0,0,1,1,1,2,2,3,3,4],

函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4。

你不需要考虑数组中超出新长度后面的元素。

---

思路：

思路一：传入vector，遍历数组删除重复元素即可
思路二：传入vector或数组，将不重复数字依次放入数组前n位

---

代码：

思路一：

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if (nums.size() == 0) return 0;

        vector<int>::iterator iter = nums.begin();
        int baseNum = *iter;
        iter++;

        while (iter != nums.end()){
            if (*iter == baseNum) {
                iter = nums.erase(iter);
            }
            else {
                baseNum = *iter;
                iter++;
            }
        }

        return nums.size();
    }
};

思路二：

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if (nums.size() == 0) return 0;

        int p = 0;
        for (int q = 1; q < nums.size(); ++q){
            if (nums[q] != nums[p]){
                nums[++p] = nums[q];
            }
        }
        return p + 1;
    }
};