leetcode之Remove Duplicates from Sorted Array II

最新推荐文章于 2025-08-14 23:25:10 发布

luckyu1

最新推荐文章于 2025-08-14 23:25:10 发布

阅读量163

点赞数

CC 4.0 BY-SA版权

文章标签： leetcode 算法面试

本文链接：https://blog.youkuaiyun.com/luckyu1/article/details/50759595

本文介绍了一种针对已排序数组的算法实现，该算法允许数组中的元素最多出现两次，并返回更新后的数组长度。通过使用双指针技巧，有效地实现了这一目标。

摘要生成于 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.

解答：

很明显的用两个指针和计数来做，用一个指针表示现在正在进行计数的数字，另一个指针表示现在处理的位置

代码很明显如下

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int start = 0;
        int end = 1;
        int size = nums.size();
        if(size <= 2)
            return size;
        int cnt = 1;
        while(end < size)
        {
            if(nums[start] == nums[end])
            {
                if(cnt < 2)
                {
                    nums[++start] = nums[end++];
                }
                else
                {
                    end++;
                }
                cnt++;
            }
            else
            {
                nums[++start] = nums[end++];
                cnt = 1;
            }
        }
        return start + 1;
    }
};