LeetCode | Remove Duplicates from Sorted Array II（删除重复的元素2）

2023年7月29日 271次阅读


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

题目解析：

最多允许两个重复，输出结果数组。


方案一：

可以另外设置O(n)的辅助空间，对每一个数据进行计数，多于2个的只输出两个，少于两个的只输出1一个。

这个方法很容易想，但不是高效的算法。要去思考如何不使用辅助空间。


方案二：

跟Remove Duplicates from Sorted Array 类似，这里用个计数变量count，当超过两个的时候，就不再移动i指针。其中有一点小细节是：开始时i和j相等，是不需要进行交换的。但是为了代码简单，忽略这种情况。只要count<=2或者arr[j] != temp的时候，都进行交换。

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        int i=0,j=0;
        int count = 0;
        int temp = A[0];
        while(j<n){
            if(A[j] == temp){   //如果和参考值相等
                if(count<=2){   //当个数少于两个的时候，就交换，无论i和j是否相等，减少分支条件
                    count++;
                    Swap(&A[i],&A[j]);
                    i++;
                }
                    count++;    //无论count是否小于2，都要进行计数
                    j++;        //都要让j++
                continue;
            }
            count = 1;          //当和参考值不相等的时候，count重新置一，并且更新参考值
            temp = A[j];
            Swap(&A[i],&A[j]);
            i++;
            j++;
        }
        return i;
    }
    void Swap(int *a,int *b){
        int tmp = *a;
        *a = *b;
        *b = tmp;
    }
};



点赞