【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;
    }
};

你可能感兴趣的:(LeetCode)