Remove Duplicates from Sorted Array II

Leetcode

题目：

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].

给定一个有序表，每个元素最多出现两次。删除掉多余的元素，返回剩余元素的个数。

思路：

两个指针，一个指向合法位置的下一个位置 （index） ，另一个用于遍历元素 (i) 。两个指针之间的是无效元素。

这个时候判断 i 指向的元素和 index - 2 指向的元素是否相同。

• 相同：说明 i 指向的元素是多余无效的。 i 后移即可。
• 不相同：说明 i 指向的元素是有效的，即出现次数不多于2次。此时，用 i 替换 index 即可。index 需要后移。

代码如下：

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        int deleted = 0; //已经删除的元素个数
        if(n < 3)        //最多含有2个元素，一定满足题意
            return n;
        int index = 2; //可能不合法的位置
        for(int i = 2; i < n; i++)
        {
            if(A[i] != A[index - 2] )
                A[index++] = A[i];
        }
        return index;
    }
};