一个数组 nums
和一个值 val
,需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,必须仅使用 O(1)
额外空间并 原地 修改输入数组。
元素的顺序可以改变。不需要考虑数组中超出新长度后面的元素。
class Solution {
public int removeElement(int[] nums, int val) {
int sum = nums.length;
int i = 0;
while(i < sum){
if(nums[i] == val){
nums[i] = nums[sum - 1];
sum--;
}else{
i++;
}
}
return sum;
}
}