leetcode/problems/80-remove-duplicates-from-sorted-array-ii.cpp at main · Sa4dUs/leetcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
// https://leetcode.com/problems/remove-duplicates-from-sorted-array-ii/

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int n = size(nums);

        if (n < 3)
            return n;

        auto it = nums.begin() + 2;
        auto aux = it;

        while (it < nums.end()) {
            if (*it != *(aux - 2)) {
                *aux = *it;
                ++aux;
            }

            ++it;
        }

        return aux - nums.begin();
    }
};