题目描述:
-
给你一个 有序数组 nums,请你 原地删除 重复出现的元素,使每个元素最多出现两次 ,返回删除后数组的新长度。
-
不要使用额外的数组空间,你必须在原地修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例一:
输入:nums = [1,1,1,2,2,3]
输出:5, nums = [1,1,2,2,3]
解释:函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。不需要考虑数组中超出新长度后面的元素。
示例二:
输入:nums = [0,0,1,1,1,1,2,3,3]
输出:7, nums = [0,0,1,1,2,3,3]
解释:函数应返回新长度 length = 7, 并且原数组的前五个元素被修改为 0, 0, 1, 1, 2, 3, 3 。 不需要考虑数组中超出新长度后面的元素。
来源:力扣(LeetCode)
题解:
class Solution {
public int removeDuplicates(int[] nums) {
if(nums.length <= 2)
return nums.length;
int index = 2;
for(int i = 2; i < nums.length; i++){
if(nums[i] != nums[index-2])
nums[index++] = nums[i];
}
return index;
}
}
使用 双指针 ,一个指针指向当前的元素,一个指针去遍历。
index 始终指向要被替换的元素。
i 用来遍历,每次循环都 + 1。
1)数组长度小于2,直接返回,因为每个元素最多可出现两次。
2)index 和 i 都从第三个(数组索引值为 2 )元素开始。
3) [ i ] 与 [ index - 2 ] 比较 ,相等则 i 往后遍历,寻找一个不相等的数,将[ index ] 替换掉。【 由于数组是有序的,如果 [ i ] 和 [ index - 2 ] 相等,那必然和中间的数也相等 ,那么就需要往后找一个和 [ index ] 不相等的数把它替换掉 】
4) i 往后遍历,寻找一个与 [ index - 2 ]不相等的数,不相等则赋给 [ index ] ( [ index ] = [ i ] ) 然后 index = index + 1,index便指向下一个可能需被替换的元素。
例: [ 1, 1, 1, 2, 2, 3]
i 一直往后遍历,一直都在寻找和 [ index - 2 ] 不相等的那个数,找到就赋给 [ index ] 。就可以保证在 index 之前是不会有三个相等的数出现的。
评论中的 java 代码最少解法,思想一样的。
class Solution {
public int removeDuplicates(int[] nums) {
int i = 0;
for (int num : nums) {
if (i < 2 || num > nums[i - 2]) {
nums[i++] = num;
}
}
return i;
}
}
for (int num : nums)
是一个foreach循环遍历,就是把nums这个数组进行遍历,它有多少个数,就遍历多少遍。遍历的时候每次就把其中的一个值给num。
因为是有序数组,直接用比较大小来判断是否是不同的值。如果大于,赋值给nums[ i ]并且 i ++。