Leetcode 80. Remove Duplicates from Sorted Array II

96
SnailTyan
2018.08.23 19:04 字数 33

文章作者:Tyan
博客:noahsnail.com  |  CSDN  |  简书

1. Description

Remove Duplicates from Sorted Array II

2. Solution

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

Reference

  1. https://leetcode.com/problems/remove-duplicates-from-sorted-array-ii/description/
Leetcode
Web note ad 1