版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_35356190/article/details/82913319
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
示例 1:
给定 nums = [1,1,1,2,2,3], 函数应返回新长度 length =5
, 并且原数组的前五个元素被修改为1, 1, 2, 2, 3 。
你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 nums = [0,0,1,1,1,1,2,3,3], 函数应返回新长度 length =7
, 并且原数组的前五个元素被修改为0, 0, 1, 1, 2, 3, 3 。
你不需要考虑数组中超出新长度后面的元素。
class Solution {
public int removeDuplicates(int[] nums) {
if(null == nums){
return 0;
}
if(nums.length<3){
return nums.length;
}
int repeatCount = 1, count = 0;
for(int i=nums.length-1;i>0;--i){
if(nums[i]==nums[i-1]){
if(1==repeatCount){
++repeatCount;
}else if(2==repeatCount){
for(int j=i+1;j<nums.length;++j){
nums[j-1]=nums[j];
}
++count;
}
}else{
repeatCount=1;
}
}
return nums.length-count;
}
}
刚开始思路不是很清晰,变量i从0遍历到数组尾部,answer error!
因为会重复数为2时,数组向前移动一位,会覆盖索引i处,覆盖原来的值,然后i继续向前遍历,会导致i值变化,而repeatCount=2,继续查训会导致repeatCount不准确。
错误例:
[1][1][1][2][2][3]
[1][2][2][2][3]
[1][1][2][3]
逆向思维,从后往前遍历,则可以避免覆盖情形出现。
二 O(n)解法
class Solution {
public int removeDuplicates(int[] nums) {
if(null == nums){
return 0;
}
if(nums.length<3){
return nums.length;
}
int i=1;
for(int j=i+1;j<nums.length;++j){
if(nums[i]!=nums[j] || (nums[j]==nums[i] && nums[i-1]!=nums[j])){
nums[i+1]=nums[j];
++i;
}
}
return i+1;
}
}
目标:一次过!最优解!