题目描述
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。
思路:
双指针:i是慢指针,j是快指针,当nums[j]与给定的值相等时,j加一,继续循环,只要nums[j]和val不相等,递增两个索引,复制nums[j]到nums[i]
【可能数组只有一个元素,该元素与val相等,新数组长度为0,因此长度初始化为0,即与i等值,返回i即可】
class Solution {
public int removeElement(int[] nums, int val) {
int i = 0;
for(int j = 0;j<nums.length;j++){
if(nums[j] != val){
nums[i] = nums[j];
i++;
}
}
return i;
}
}