leetcode80 Remove Duplicates from Sorted Array II

最新推荐文章于 2025-08-04 20:35:07 发布

怒吃树

最新推荐文章于 2025-08-04 20:35:07 发布

阅读量208

点赞数

文章标签： leetcode

本文探讨了数组中允许元素最多重复两次的去重问题，并提供了两种解决方案：一种是使用LinkedHashMap来记录元素出现次数的方法；另一种是通过巧妙地检查相隔一个的元素是否不同来进行更新的算法。

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

自己的思路是通过linkedhashmap存储每个元素出现的次数，当出现大于2时长度减一，然后遍历map赋值返回length。

public int removeDuplicates(int[] nums) {
    int length=nums.length;
    Map<Integer,Integer> map=new LinkedHashMap<>();
    for(int i=0;i<nums.length;i++)
    {
        if(map.get(nums[i])==null)
            map.put(nums[i],1);
        else if(map.get(nums[i])==1)
            map.put(nums[i],2);
        else if(map.get(nums[i])==2)
            length--;
    }
    int k=0;
    for(Integer i:map.keySet())
    {
        int num=map.get(i);

        while(num-->0)
        {
            nums[k++]=i;
        }
    }
    return length;
}

网上看到一个巧妙的解法，

题意为允许一个重复，所以判断就不是相邻而是相隔一个的数字是否不一样。

不一样就需要更新值，但注意到这题更新数组不能是当前的状态，而需要是更新前一个不同的数，用temp存当前值以便于下一次赋值。

public int removeDuplicatesII(int[] nums)
{
    if(nums.length==0) return 0;
    if(nums.length==1) return 1;
    int num=1,temp=nums[1];
    for(int i=2;i<nums.length;i++)
    {
        if(nums[i]!=nums[i-2])
        {
            nums[num++]=temp;
            temp=nums[i];
        }
    }
    nums[num++]=temp;
    return num;


}