[LeetCode]Remove Duplicates from Sorted Array II

CiaoLiang

于 2015-12-19 20:47:47 发布

阅读量369

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/CiaoLiang/article/details/50359626

LeetCode 专栏收录该内容

149 篇文章

订阅专栏

本文介绍了一个数组去重的算法实现，允许元素重复出现两次。通过双指针技术，该算法可以有效地将数组中的重复元素数量控制在最多两次，并返回处理后的有效长度。

摘要生成于 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 ret = 1;
        int p1 = 0;
        int p2 = 0;
        int count = 1;
        if(nums.size()<2)
            return nums.size();
        nums.push_back(INT_MAX);
        for(p2=1; p2<nums.size(); ++p2){
            if(nums[p2]==nums[p2-1]){
                count++;
            }
            else{
                int add = 1;
                if(count>=2) add = 2;
                ret += add;
                count=1;
                for(int i=p1; i<p1+add; ++i)
                    nums[i] = nums[p2-1];
                p1 += add;
            }
        }
        return ret-1;
    }
};