LeetCode 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.
• 地址

问题分析

• 该题是LeetCode 26. Remove Duplicates from Sorted Array的进阶题目，原题是重复元素只保留一个，比如对于 [1,1,2,2,2,3]，去重之后是[1,2,3] ;该题是对于重复元素保留2个,去重之后是[1,1,2,2,3]。依旧是用 [0…k)来保存去重之后的结果，那么当遍历到 nums[i] 时，只要 nums[i] ！= nums[k - 2],说明nums[i] 在[0..k)中出现次数尚未超过两次，可以将nums[i] 放在 k 处。总结如下：
  
  • 若 k < 2,那么nums[i] 可以直接放在k处
  • 若 k >= 2,只有nums[i] ！= nums[k - 2]，才可以将nums[i] 放在 k 处

经验教训

• 如果题目要求。对于重复元素保留 x 个呢，如何处理?
  
  • 若 k < x,那么nums[i] 可以直接放入k处
  • 若 k >= x, 只有nums[i] ！= nums[k - x]，才可以将nums[i] 放在 k 处

代码实现

    public int removeDuplicates(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }
        int k = 0;
        for (int i = 0; i < nums.length; i++) {
            if (k < 2 || nums[i] != nums[k - 2]) {
                nums[k++] = nums[i];
            }
        }
        return k;
    }