80. Remove Duplicates from Sorted Array II(python+cpp)

本文介绍了一种在原地修改数组以去除重复项的算法,确保每个元素最多出现两次,并详细解析了Python和C++的实现方法。

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

题目:

Given a sorted array nums, remove the duplicates in-place such that duplicates appeared at most twice and return the new length.
Do not allocate extra space for another array, you must do this by modifying the input array in-place with O(1) extra memory.
Example 1:

Given 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 respectively.
It doesn't matter what you leave beyond the returned length. 

Example2:

Given nums = [0,0,1,1,1,1,2,3,3],
Your function should return length = 7, with the first seven elements of nums being modified to 0, 0, 1, 1, 2, 3 and 3 respectively.

It doesn’t matter what values are set beyond the returned length.
Clarification:
Confused why the returned value is an integer but your answer is an array?
Note that the input array is passed in by reference, which means modification to the input array will be known to the caller as well.
Internally you can think of this:

// nums is passed in by reference. (i.e., without making a copy) 
int len = removeDuplicates(nums);
// any modification to nums in your function would be known by the caller. 
// using the length returned by your function, it prints thefirst len elements. 
for (int i = 0; i < len; i++) {
    print(nums[i]); 
}

解释:
本来以为是类似于82. Remove Duplicates from Sorted List II(python+cpp)的要求,结果发现不是的。
题目要求:给定一个排好序的数组,in-place删除一些重复元素使得最终返回的数组多种每个数组至多出现两次。返回最终的数组的长度。
其实这种题目也是需要用两个指针的,先写一个不那么明显的用指针的方法,速度比较慢。
pytho代码:

class Solution:
    def removeDuplicates(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        i=0
        for n in nums:
            if i<2 or n>nums[i-2]:
                nums[i]=n
                i+=1
        return i

c++代码:

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        //j指向当前需要赋值的位置
        int j=0;
        for(int i=0;i<nums.size();i++)
        {
            if (i<2 || nums[j-2]<nums[i])
            {
                nums[j]=nums[i];
                j+=1;
            }
        }
        return j;
    }
};

用两个指针的解法:
python代码:

class Solution:
    def removeDuplicates(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        n=len(nums)
        if n<=2:
            return n
        count,i,j=1,1,1
        while i< n:
            if nums[i]==nums[i-1]:
                if count<2:
                    count+=1
                    nums[j]=nums[i]
                    j+=1
                i+=1
            else:
                count=1
                nums[j]=nums[i]
                i+=1
                j+=1
        return j     

c++代码:

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int n=nums.size();
        if(n<=2)
            return n;
        int count=1,i=1,j=1;
        while(i<n)
        {
            if(nums[i]==nums[i-1])
            {
                if (count<2)
                {
                    count++;
                    nums[j]=nums[i];
                    j++;
                }
            }
            else
            {
                count=1;
                nums[j]=nums[i];
                j++; 
            }
            i++;
        }
        return j;
    }
};

总结:
注意变量的初始化和一些特殊情况的判断。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值