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有序数组中出去重复项,这里运行最多重复次数是二次,因此我们需要定义一个count来记录重复次数,count初始化为1,当出现一次时count减1,下次再重复出现时,快指针直接前进一步,如果这时不是重复的则count恢复为1,该数组是有序的,所以一旦不出现重复的数,后面的数一定比这个数大。
int removeDuplicates(int* nums, int size) {
if (size <= 2) {
return size;
}
int i = 0;
int j = 1;
int count = 1;
while (j<size)
{
//当nums[i]== nums[j],count == 0时,此时这个数是第一次和后面的一个数相等
if (nums[i] == nums[j] && count == 0) {
++j;
}
else
{ //此时判断是否与后面的数还能相等,相等时就 --count,
//否则不相等count还是1,说明这个数只出现两次
if (nums[i] == nums[j]) {
--count;
}
else
{
count = 1;
}
//此时我们让第二次出现的这个数后移动一位,并放在这个数后面
nums[++i] = nums[j++];
}
}
return i+1;
}