LeetCode OJ 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.

---

【题目分析】

在前一个题目的基础上，这个题目稍微做了变化。上一个题目中每一个数字只能出现一次，在这个题目中每一个数字允许出现两次。

---

【思路】

回忆一下，在Remove Duplicates from Sorted Array这个题目中，一个巧妙的方法就是设置了两个下标 i 和 j，i 用来遍历数组，j 用来指示数组中不重复元素最后一个元素的位置。那么同样在这个题目中我们也设置这样两个变量，指示此时的 j 用来指示数组中重复次数不超过2的元素最后的位置。我们怎么知道一个元素重复出现的次数呢？难道要设置一个变量来记录当前元素重复出现了多少次吗？一个很巧妙的办法就是比较当前元素nums[i] 是否和前前一个元素 nums[j-1] 相同，如果不相同则nums[++j] = nums[i]，否则的话当前元素出现次数肯定是大于2次了，继续向后遍历数组即可。

这个过程如下：

• j = 1; i = 2;

• num[i] 等于 nums[j -1]; i++;

• nums[i] 不等于 nums[j -1]; nums[++j] = nums[i]; i++;

• num[i] 不等于 num[j -1]; nums[++j] = nums[i]; i++;

• num[i] 等于 num[j -1] ; i++;

• num[i] 不等于 num[j -1]; nums[++j] = nums[i];

---

【java代码】

public class Solution {
    public int removeDuplicates(int[] nums) {
        if (nums.length <= 2)
            return nums.length;
        int j = 1;
        for(int i=2; i<nums.length; i++) {
        if (nums[j-1] != nums[i])
            nums[++j] = nums[i];
        }
        return j+1;
    }
}