leetcode80 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.

    自己的思路是通过linkedhashmap存储每个元素出现的次数，当出现大于2时长度减一，然后遍历map赋值返回length。

public int removeDuplicates(int[] nums) {
int length=nums.length;
Map<Integer,Integer> map=new LinkedHashMap<>();
for(int i=0;i<nums.length;i++)
{
if(map.get(nums[i])==null)
map.put(nums[i],1);
else if(map.get(nums[i])==1)
map.put(nums[i],2);
else if(map.get(nums[i])==2)
length--;
}
int k=0;
for(Integer i:map.keySet())
{
int num=map.get(i);

while(num-->0)
{
nums[k++]=i;
}
}
return length;
}

    网上看到一个巧妙的解法，

题意为允许一个重复，所以判断就不是相邻而是相隔一个的数字是否不一样。

不一样就需要更新值，但注意到这题更新数组不能是当前的状态，而需要是更新前一个不同的数，用temp存当前值以便于下一次赋值。

public int removeDuplicatesII(int[] nums)
{
if(nums.length==0) return 0;
if(nums.length==1) return 1;
int num=1,temp=nums[1];
for(int i=2;i<nums.length;i++)
{
if(nums[i]!=nums[i-2])
{
nums[num++]=temp;
temp=nums[i];
}
}
nums[num++]=temp;
return num;

}