必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成
所以只能修改原数组。
直接对满足 nums[i] == val
的值进行依次覆盖即可。
public int removeElement(int[] nums, int val) {
int idx = 0;
for (int i = 0; i < nums.length; i++) {
if (nums[i] != val) {
nums[idx] = nums[i];
idx++;
}
}
return idx;
}
元素的顺序可以改变
因为对顺序没有要求,所以可以直接使用数组尾部的值覆盖前面的值。
class Solution {
public int removeElement(int[] nums, int val) {
int idx = 0;
int last = nums.length;
while (idx < last) {
if (nums[idx] == val) {
nums[idx] = nums[last - 1];
last--;
} else {
idx++;
}
}
return idx;
}
}
- <End /> -