LeetCode Remove Duplicates from Sorted Array II

最新推荐文章于 2020-02-08 17:36:31 发布

原创最新推荐文章于 2020-02-08 17:36:31 发布 · 462 阅读

0 ·

CC 4.0 BY-SA版权

OJ 专栏收录该内容

986 篇文章

订阅专栏

本文介绍了一种处理有序数组的方法，确保重复元素最多出现两次。通过遍历数组并计数，该算法有效地调整了数组内容并返回了新长度。

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

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

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

Your function should return length = 5, with the first five elements of nums being 1, 1, 2, 2 and 3. It doesn't matter what you leave beyond the new length.

题意：给出一有序数组，处理后使得重复的元素不能超过2个，并且返回数组的元素个数

思路：

算法步骤如下

1、如果当前元素与前一个元素相等，并且重复的个数没有超过2，就将当前元素存入数组，并且将重复记数加1

2、如果当前元素与前一个不相等，就将当前元素存入数组，重复记数置为1

代码如下

class Solution
{
    public int removeDuplicates(int[] nums)
    {
        int len = nums.length;
        int pos = 1;
        int cnt = 1;
        int ret = len > 0 ? 1 : 0;

        for (int i = 1; i < len; i++)
        {
            int cur = i;
            while (cur < len && nums[cur] == nums[cur - 1])
            {
                if (cnt < 2)
                {
                    nums[pos++] = nums[cur];
                    ret++;
                }
                cnt++;
                cur++;
            }

            if (cur < len)
            {
                nums[pos++] = nums[cur];
                cnt = 1;
                ret++;
            }
            i = cur;
        }
       return ret;
    }
}