leetcode80. Remove Duplicates from Sorted Array II

大大kc

于 2017-06-17 23:04:16 发布

阅读量233

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode twopointer array

本文链接：https://blog.youkuaiyun.com/rxt2012kc/article/details/73394746

leetcode 专栏收录该内容

187 篇文章

订阅专栏

本文介绍了一种改进的数组去重算法，允许每个元素最多出现两次。提供了两种解决方案：一种是通过计数累加的方式实现，另一种是通过比较当前元素与前两个元素来决定是否保留。这些方法适用于已排序的数组。

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

80. Remove Duplicates from Sorted Array II

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.

解法一

相同元素的个数依次累加。注意从1开始。

public class Solution {
    public int removeDuplicates(int[] nums) {
        if(nums.length == 0 && nums == null) {
            return 0;
        }

        int i = 1, j = 1;
        int count = 1;
        while (j < nums.length) {
            if (nums[j] != nums[j - 1]) {
                count = 1;
                nums[i++] = nums[j];
            } else {
                if (count < 2) {
                    nums[i++] = nums[j];
                    count++;
                }
            }
            j++;
        }

        return i;
    }
}

解法二

判断i-2元素是否小于该元素，小于就把该元素添加到数组里。

public class Solution {
    public int removeDuplicates(int[] nums) {
        if (nums == null || nums.length == 0) {
           return 0;
        }

        int i = 0;
        for (int num : nums) {
            if (i < 2 || nums[i - 2] < num) {
                nums[i++] = num;
            }
        }

        return i;
    }
}