80. Remove Duplicates from Sorte

题目链接

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

代码

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int count = 1;
        for (int i = nums.size() - 2; i >= 0; --i) {
            if (nums[i] == nums[i + 1]) {
                if (count >= 2) {
                    nums.erase(nums.begin() + i);
                    continue;
                } else {
                    count++;
                }
            } else {
                count = 1;
            }
        }
        return nums.size();
    }
};