从排序阵列中删除重复 II

题目来源

Remove Duplicates from Sorted Array II

问题描述

“删除重复项目” 的进阶：
如果重复最多被允许两次，又该怎么办呢？

例如：
给定排序数列 nums = [1,1,1,2,2,3]

你的函数应该返回长度为 5，nums 的前五个元素是 1, 1, 2, 2 和 3。

解决方案

C++

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int k = 1;//只允许两次
        int size = nums.size();
        if (size == 0 || size == 1)
            return size;
        int pos = 0;
        for (int i = 1; i < size; i++)
        {
            if (nums[i] != nums[pos])
            {
                k=1;
                pos++;
                nums[pos] = nums[i];            
            }
            else if (k<2){  
                k++;
                pos++;
                nums[pos] = nums[i];
            }
        }
        return pos+1; 
    }
};

Java

class Solution {
    public int removeDuplicates(int[] nums) {
        if (nums.length <= 2)
            return nums.length;
        int index = 2;
        for (int i = 2; i < nums.length; i++) { //从第三个元素开始
            if (nums[i] != nums[index - 2]) //由于排好了序，相等说明 i至少是第三个重复元素
            {
                nums[index] = nums[i];
                index++;
            }
        }
        return index;
    }
}