题目:
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.
题意及分析:给出一个升序数组,删除其中的重复元素,删除之后每个元素最多只能出现两次
代码1:记录当前有多少个连续数,若小于等于2,则直接写入,若大于2,则只写入最前面的两个数。对数组最后面的数需要进行单独判断
class Solution { public int removeDuplicates(int[] nums) { if(nums.length==0) return 0; int res = 0; int count = 1; //记录连续的数个数 for(int i=0;i<nums.length;i++){ if(i>0 && nums[i]==nums[i-1]) { count++; if(count>2) continue; //大于两次直接跳过 } else{ //当前数不等于前面的数,重置为1 count=1; } nums[res++] = nums[i]; } return res ; } }
代码二:(1)如果i<2那么直接添加进数组,(2)否则如果当前数在之前已经存在两个,那么跳过,否则添加进数组
class Solution { public int removeDuplicates(int[] nums) { int i = 0; for (int n : nums) if (i < 2 || n > nums[i - 2]) nums[i++] = n; return i; } }