题目:
给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
思路 :双指针。前半段是有效部分,存储的是不等于val的元素;后半段是无效部分,存储的是等于val的元素。最终答案返回有效部分的结尾下标。
代码:
java
class Solution {
public int removeElement(int[] nums, int val) {
int j = nums.length - 1; // j指向不等于val的最靠后的位置
for (int i = 0; i <= j; i++) {
if (nums[i] == val) {
nums[i] = nums[j];
nums[j] = val;
j--;
i--;
}
}
return j + 1;
}
}
性能:时间复杂度 O(n) 空间复杂度 O(1)