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.

題意:

這題是Remove Duplicates from Sorted Array的延伸，而在這一題與上一題不同的是：每個元素可以重複一次，例如：

[1,1,1,2,3]->[1,1,2,3]

題解:

這題是將數組中一樣的元素去除，故有兩個指針:

1. 指向目前元素(i)
2. 指向目前沒有重複的元素(j)

但要加上三個限制條件:
1.若元素不重複(進行指針移動)
則numOfElements為1
nums[j] = nums[i]
j ++
2.若元素重複，且numOfElements為1 (進行指針移動)
則numOfElements為2
nums[j] = nums[i]
j ++
2.若元素重複，且numOfElements為2級以上
跳過這回合
    ex:
         1,1,1,2,3,4,5
step1 i=1,j=0
         1,1,1,2,3,4,5
step2 i=2,j=1
         1,1,1,2,3,4,5
step3 i=3,j=1
         1,1,1,2,3,4,5 
step4 i=4,j=2
         1,1,2,2,3,4,5 
step5 i=5,j=3
         1,1,2,3,3,4,5 
step6 i=6,j=4
         1,1,2,3,4,4,5
step7 i=7,j=5
         1,1,2,3,4,5,5 
step8 i=8,j=6
         1,1,2,3,4,5,5 

package LeetCode.Medium;
public class RemoveDuplicatesFromSortedArrayII {
    /*
這題是將數組中一樣的元素去除，故有兩個指針:
1.指向目前元素(i)
2.指向目前沒有重複的元素(j)

但要加上三個限制條件:
1.若元素不重複
	則numOfElements為1
	nums[j] = nums[i]
	j ++
2.若元素重複，且numOfElements為1
	則numOfElements為2
	nums[j] = nums[i]
	j ++
2.若元素重複，且numOfElements為2級以上
	跳過這回合
    ex:
      1,1,1,2,3,4,5
step1 i=1,j=0
      1,1,1,2,3,4,5
step2 i=2,j=1
      1,1,1,2,3,4,5
step3 i=3,j=1
      1,1,1,2,3,4,5 
step4 i=4,j=2
      1,1,2,2,3,4,5 
step5 i=5,j=3
      1,1,2,3,3,4,5 
step6 i=6,j=4
      1,1,2,3,4,4,5
step7 i=7,j=5
      1,1,2,3,4,5,5 
step8 i=8,j=6
      1,1,2,3,4,5,5 
    */
    public int removeDuplicates(int[] nums) {
        if(nums == null || nums.length == 0)
            return 0;
        
        if(nums.length == 1)
            return 1;
        
        int numOfElements = 1;
        int n = nums.length;
        int j = 0;
        
        //歷遍數組所有元素(這種題可以從統一從1比較)
        for(int i = 1; i < n; i ++) {
            if(nums[i] == nums[j] && numOfElements > 1) {
                numOfElements ++;
                continue;
            }
            
            if(nums[i] == nums[j] && numOfElements == 1) {
                j ++;
                nums[j] = nums[i];
                numOfElements ++;
                continue;
            }
            
            if(nums[i] != nums[j]) {
                j ++;
                nums[j] = nums[i];
                numOfElements = 1;
            }
        }
        
        return j + 1;
    }
}