每天一道LeetCode-----从有序数组中删除重复元素

Remove Duplicates from Sorted Array

原题链接Remove Duplicates from Sorted Array

给定一个有序数组，对于数组中的重复项，只需要保留一个，其余的都删掉

和每天一道LeetCode—–移除有序链表中的重复节点类似

只需要记录当前找到的元素，当再遍历到一个元素时，如果和之前找到的元素一样，就丢掉

代码如下

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int prev = 0;
        for(auto& n : nums)
        {
            /* prev - 1是上一个找到的元素，如果和它相等，就丢掉 */
            if(prev == 0 || n != nums[prev - 1])
                nums[prev++] = n;
        }
        return prev;
    }
};

---

Remove Duplicates from Sorted Array II

原题链接Remove Duplicates from Sorted Array II

允许保留两个重复元素

令prev指向下一个元素应该存放的位置，初始为0，假设当前遍历到一个元素，那么它前面两个位置分别是prev - 1和prev - 2

考虑nums[prev-2]，如果当前遍历到的元素和nums[prev-2]相等，说明是第三个重复元素，应该丢掉。如果当前遍历到的元素大于nums[prev-2]（因为递增），那么就可以将其添加到结果中

代码如下

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int prev = 0;
        for(auto& n : nums)
        {
            /* 开始两个可以添加，后面的需要和它前面两个比较 */
            if(prev < 2 || n > nums[prev-2])
                nums[prev++] = n;
        }
        return prev;
    }
};

---

这两道题主要需要想到如果判断当前遍历到的元素是否满足条件，对于只保留一个的，那么和nums[prev-1]比较即可，对于保留两个的，那么只需要和nums[prev-2]比较。因为prev指向的是当前遍历到的元素应该插入的位置，所以nums[prev-1]和nums[prev-2]分别表示它之前的元素和之前的之前的元素:)