描述:
给你一个 升序排列 的数组
nums
,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。
示例:
输入:
nums
= [0,0,1,1,1,2,2,3,3,4]
输出:5,nums
= [0,1,2,3,4]
解释:函数应该返回新的长度 5 , 并且原数组nums
的前五个元素被修改为 0, 1, 2, 3, 4 。不需要考虑数组中超出新长度后面的元素。
注意点:
不要使用额外的空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
思路:
使用双指针prev
、next
遍历数组,next
依次向前遍历,确定去重数组的下一项。
-
如果右指针指向的值等于左指针指向的值,左指针不动。
-
如果右指针指向的值不等于左指针指向的值,那么左指针往右移一步,然后再把右指针指向的值赋给左指针。
提示:遍历过程中,数组从0到
prev
为去重数组
代码:
public static int removeDuplicates(int[] nums) {
int prev = 0, next = 1;
while (next < nums.length) {
if (nums[prev] != nums[next]) {
nums[++prev] = nums[next];
}
next++;
}
return prev + 1;
}
收获:
-
循环次数确定使用while循环,不缺定使用for循环
-
assert语句只用于检查,不参与程序的实际的运行,使用时需要在idea中手动开启,使用方法:
asser condition:expr;
-
数组的输出可以使用Arrays工具类的toString方法,憨憨的我每次使用for循环遍历输出,除此之外,Arrays工具类还有许多常用方法,sort,copyOf,binarySearch,equals,fill等方法,需要注意的是,sort方法默认升序,如果想实现Comparator接口自定义排序规则,数组必须为引用类型
-
算法还可以进行优化,当数组中无重复元素时,
prev
和next
指向的值会从头到尾交换一边。可以添加条件,当prev
与next
距离大于1时再交换数值。注意点:
prev++;
的位置,我之前放到了判断距离语句的上面,一直报错,又花了多一段时间才明白
public static int removeDuplicates(int[] nums) {
int prev = 0, next = 1;
while (next < nums.length) {
if (nums[prev] != nums[next]) {
if ((next - prev) > 1) {
nums[prev + 1] = nums[next];
}
prev++;
}
next++;
}
return prev + 1;
}