[LeetCode] Remove Duplicates from Sorted Array II

原创于 2014-06-16 09:46:19 发布 · 789 阅读

0 ·

CC 4.0 BY-SA版权

算法同时被 2 个专栏收录

43 篇文章

订阅专栏

面试

40 篇文章

订阅专栏

这是LeetCode中的一道问题，要求解决在允许最多重复两次的情况下，从已排序数组中移除重复项。给定一个有序数组A，原地修改数组并返回新的数组长度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

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

解答：

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if(n <= 1) {
            return n;
        }
        int num = 1;
        int cur = 0;
        for(int i = 1; i < n; i++) {
            if(A[i] == A[cur]) {
                if(num < 2) {
                    A[++cur] = A[i];
                    num++;
                }
                else {
                    continue;
                }
            }
            else {
                num = 1;
                A[++cur] = A[i];
            }
        }
        return cur + 1;
    }
};

思路：相比“Remove Duplicates from Sorted Array”，在遍历的过程中加了一个计数器，保证最多重复次数为2.