80. Remove Duplicates from Sorted Array II

原创于 2016-04-19 17:21:49 发布 · 192 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #java

leetcode(java实现) 专栏收录该内容

210 篇文章

订阅专栏

本文介绍了一种解决方案，用于处理一个排序数组，使得数组中每个元素出现次数最多为两次，同时返回处理后数组的有效长度。通过使用布尔标志数组跟踪每个元素的出现次数，并将符合条件的元素复制到新数组，最后再将该数组的内容覆盖原数组。

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.

代码;用一个额外的数组，记录下重复次数不超过2次的部分，然后再赋值给nums

代码如下(已通过leetcode)

public class Solution {
public int removeDuplicates(int[] nums) {
if(nums.length==0) return 0;
boolean[] flag=new boolean[nums.length];
int count=1;
int sum=1;
for(int i=1;i<nums.length;i++) {
if(nums[i]==nums[i-1]) {
count++;
if(count<3) sum++;
else flag[i]=true;
} else {
count=1;
sum++;
}
}
int[] temp=new int[sum];
int j=0,i=0;
while(i<nums.length&&j<sum) {
if(!flag[i]) {
temp[j]=nums[i];
i++;
j++;
} else i++;
}

for(int k=0;k<sum;k++)
nums[k]=temp[k];
return sum;
}
}