【LeetCode】80. Remove Duplicates from Sorted Array II (2 solutions)

Remove Duplicates from Sorted Array II

Follow up for "Remove Duplicates":
What if duplicates are allowed at most twice?

For example,
Given sorted array A = [1,1,1,2,2,3],

Your function should return length = 5, and A is now [1,1,2,2,3].

解法一：

我是Pick One!的，所以一开始没关心"Remove Duplicates"中的要求，不知道是排好序的，直接开始做了。

代码比较简单，建立map，存放每个元素与相应的出现次数。

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        int sum = 0;
        map<int,int> m;
        vector<int> v;
        for(int i = 0; i < n; i ++)
        {
            map<int,int>::iterator it = m.find(A[i]);
            if(it == m.end())
            {
                m.insert(map<int,int>::value_type(A[i], 1));
                v.push_back(A[i]);
                sum ++;
            }
            else if(it->second == 1)
            {
                it->second ++;
                v.push_back(A[i]);
                sum ++;
            }
            else
            {
                it->second ++;
            }
        }
        for(vector<int>::size_type st = 0; st < v.size(); st ++)
            A[st] = v[st];
        return sum;
    }
};

解法二：

后来发现是排好序的，那么使用in-place做法就可以了。

使用index记录新的A数组下一个的位置，使用i扫描原始A数组。

核心思想就是：

如果i扫到的当前元素在index之前已经存在两个（注意，由于A是排好序的，因此只需要判断前两个就行），

那么i继续前进。否则将i指向的元素加入index，index与i一起前进。

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if(n < 3)
            return n;
        int index = 2;
        for(int i = 2; i < n; i ++)
        {
            if(A[i] != A[index-2])
                A[index ++] = A[i];
        }
        return index;
    }
};