LeetCode(二)------Remove Duplicates from Sorted Array II

最新推荐文章于 2020-03-25 14:33:59 发布

zhumqs

最新推荐文章于 2020-03-25 14:33:59 发布

阅读量143

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签：数组去重复 leetcode

本文链接：https://blog.youkuaiyun.com/DjokerMax/article/details/81506183

LeetCode 专栏收录该内容

50 篇文章

订阅专栏

本文探讨了在有序数组中去除重复元素的问题，特别是当元素最多允许出现两次的情况。通过使用一个额外的变量跟踪元素出现的次数，可以有效地在O(n)的时间复杂度和O(1)的空间复杂度下解决问题。

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

描述：
Follow up for “Remove Duplicates”: What if duplicates are allowed at most twice?
For example, given sorted array A = [1,1,1,2,2,3] , your function should return length = 5 , and A isnow [1,1,2,2,3]
题意：
同样是有序数组去重复问题，只不过现在每个元素最多可出现两次。
思路：
加一个变量记录一下元素出现的次数即可。这题因为是已经排序的数组，所以一个变量即可解决。如果是没有排序的数组，则需要引入一个hashmap来记录出现次数。
实例代码1：

// Time complexity: O(n), Space Complexity: O(1)
public class Solution {
    public int removeDuplicates(int[] nums) {
        if (nums.length <= 2) return nums.length;
        int index = 2;
        for (int i = 2; i < nums.length; i++){
            if (nums[i] != nums[index - 2])
            nums[index++] = nums[i];
        } 
        return index;
    }
}

上面的代码略长，不过扩展性好一些，例如将 occur < 2 改为 occur < 3 ，就变成了允许重复最多3次。下面是一个更简洁的版本：

// Time Complexity: O(n), Space Complexity: O(1)
public class Solution {
    public int removeDuplicates(int[] nums) {
        int n = nums.length;
        int index = 0;
        for (int i = 0; i < n; ++i) {
            if (i > 0 && i < n - 1 && nums[i] == nums[i - 1] && nums[i] == nums[i + 1])
                continue;
            nums[index++] = nums[i];
        } 
        return index;
    }
}