Remove Duplicates from Sorted Array II

https://oj.leetcode.com/problems/remove-duplicates-from-sorted-array-ii/

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 is now [1,1,2,2,3].

解题思路：

与Remove Duplicates from Sorted Array相比，这道题稍微复杂点。因为要求一个数字最多可以出现两次，很容易想到在上题的基础上，加入一个变量计数。i作为遍历游标，遇到比record大的数字时，操作不变，同时将count重置为1。区别就是，遇到和record相等的数字，原来是不需要操作的，这里首先count必须++。其次，如果在++前，count<2，那么还需要把当前数字和swapIndex的数字交换位置，同时swapIndex++。这种情况下的swap开始忘记了，导致程序一直出错。

同样，不swap，直接A[swapIndex] = A[i]，也是可以的。

public class Solution {
    public int removeDuplicates(int[] A) {
        if(A.length == 0){
            return 0;
        }
        
        int swapIndex = 1;
        int count = 1;
        int record = A[0];
        
        for(int i = 1; i < A.length; i++){
            if(A[i] == record){
                if(count < 2){
                    int temp = A[swapIndex];
                    A[swapIndex] = A[i];
                    A[i] = temp;
                    swapIndex++;
                }
                count++;
            }
            if(A[i] > record){
                record = A[i];
                // if(count > 2){
                    int temp = A[swapIndex];
                    A[swapIndex] = A[i];
                    A[i] = temp;
                // }
                swapIndex++;
                count = 1;
            }
        }
        return swapIndex;
    }
}

update 2015/05/20:

二刷。压根就不要看大小，大小只用来计算出现的次数。只要次数<=2的，都要甩到前面去。

public class Solution {
    public int removeDuplicates(int[] nums) {
        if(nums.length == 0) {
            return 0;
        }
        int len = 1, count = 1, pre = nums[0];
        for(int i = 1; i < nums.length; i++) {
            if(nums[i] == pre) {
                count++;
            } else {
                count = 1;
                pre = nums[i];
            }
            if(count <= 2) {
                nums[len] = nums[i];
                len++;
            }
        }
        return len;
    }
}