给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
对于这道题我们一般采用双指针解法,并且这道题双指针也有两种解法
解法一
使用两个指针,左指针指向0处位置,右指针指向数组长度 -1 处,从左指针开始遍历,结束条件是left <= right,这里 = 也是需要判断的,当左指针处的值等于目标值时,交换左右指针的值,并且右指针--,此时左指针不动,因为左指针此时变为了之前右指针的值,需要再次判断,不等于则左指针++
代码如下
public int removeElement(int[] nums, int val) {
int n = nums.length;
int l = 0;
int r = n - 1;
while (l <= r) {
if (nums[l] == val) {
swap(nums, l, r);
r--;
}else {
l++;
}
}
return l;
}
private void swap(int[] nums, int l, int r) {
int t = nums[l];
nums[l] = nums[r];
nums[r] = t;
}
解法二
解法二挺巧妙,没有去做交换,而是使用后面值去覆盖当前值
思路如下
使用两个指针,左指针指向需要覆盖的位置,右指针指向不等于val的下标,当右指针处的值等于val时,右指针++,此时左指针不变,当右指针的值不等于val时,使用右指针处的值去覆盖左指针处的值,左指针++,最后左指针就是返回的值
代码如下
public int removeElement(int[] nums, int val) {
int n = nums.length;
int l = 0;
for (int i = 0; i < n; i++) {
if (nums[i] != val) {
nums[l++] = nums[i];
}
}
return l;
}
在26. 删除有序数组中的重复项题目中,我们也可以采用解法二的思想,不过代码有点不同,细想一下即可
当右指针的值等于左指针时,右指针++,否则左指针++
代码如下
public int removeDuplicates(int[] nums) {
int r = 0;
for (int i = 0; i < nums.length; i++) {
if (nums[i] != nums[r]) {
nums[++r] = nums[i];
}
}
return r + 1;
}