0%

刷算法(6)-移除元素

27. 移除元素

给定一个数组nums和一个值val,你需要原地移除所有数值等于val的元素,返回移除后数组的新长度。不要使用额外的数组空间,你必须在原地修改输入数组并在使用O(1) 额外空间的条件下完成。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

解法

1
2
3
4
5
6
7
8
9
10
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;
}

题解

因为题目说要原地移除,那一定是原地遍历一遍直接赋值。采用快慢指针方式,慢指针指向修正后数组索引,快指针进行遍历。这个题关键点是要找到如果遍历值和val相等则继续遍历,直到不相等进行赋值操作。