题目1:
给你一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,并返回移除后数组的新长度。要求: 不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并原地修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素.
这里使用快慢指针的做法,快指针在遍历查看元素是否合适,合适就挑选出来。慢指针用于确定元素,接受快指针传过来的值,接受了后往前移。如图:
代码如下:
public static int removeVal(int[] arr, int size, int val) {
int slow = 0;
for (int fast = 0; fast < size; fast++) {
if (arr[fast] != val) {
arr[slow] = arr[fast];
slow++;
}
}
//返回的slow就是新数组size
return slow;
}
题目2:
给你一个有序数组 nums,请你原地删除重复出现的元素,使每个元素只出现一次,返回删除后数组的新长度。不要使用额外的数组空间,你必须在原地修改输入数组 并在使用 0(1)额外空间的条件下完成。
这题用双指针,也是一样的做法,快指针遍历确定可以筛选的元素,慢指针接受快指针传过来的元素,然后慢指针往前走一步。
public static int removeDuplicate(int[] arr, int size) {
int slow = 1;
for (int fast = 0; fast + 1 < size; fast++) {
if (arr[fast] != arr[fast + 1]) {
arr[slow] = arr[fast];
slow++;
}
}
//返回的slow就是新数组size
return slow;
}
注意 slow 初始值为 1,第一个元素无论是否重复,都要被保留的。也避免了数组元素全部相同的情况。