给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 nums = [0,1,2,2,3,0,4,2], val = 2,
函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
注意这五个元素可为任意顺序。
你不需要考虑数组中超出新长度后面的元素。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-element
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
这道题,我想了很久,想不出怎么解,当一看到不能使用额外的数组空间,感觉就怂了很多,然后看了一下官方的解答,发现,这也太秀了吧,我怎么没想到呢~
-
思路:
用一个数组来解释说明 int[] nums = {0,1,2,2,3,0,4,2}; val = 2
用两个指针来记录,慢指针 i 快指针 j
遍历数组nums, 当nums[ j ] != val时,就执行 nums[ i ] = nums[ j ],这里是什么意思呢,其实就是重新赋值一下数组元素,当nums[ j ] == val 时,就跳开这个nums[ j ],执行下一轮循环,值到 j 角标到达 数组末尾 ,结束循环;执行过程:
当 j 角标为0 时, nums[ j ] 值为 0,不等于 val 值 2,即 nums[ j ] 值 0 再赋值给自己 nums[ i ],此时 i = 0
当 j 角标为1 时, nums[ j ] 值为 1,不等于 val 值 2,即 nums[ j ] 值 1 再赋值给自己 nums[ i ],此时 i = 1
当 j 角标为2 时, nums[ j ] 值为 2,等于 val 值 2,则执行下一个循环,此时 i = 2
当 j 角标为3 时, nums[ j ] 值为 2,等于 val 值 2,则执行下一个循环,此时 i = 2 , 但 j 已经是3了
当 j 角标为4 时, nums[ j ] 值为 3,不等于 val 值 2,即 nums[ j ] 值 3 赋值给 nums[ i ],此时 i = 2
原数组元素 int[] nums = {0,1,2,2,3,0,4,2}; val = 2
到目前的步骤后数组内元素为就是int[] nums = {0,1,3,2,3,0,4,2}; val = 2
就是后面的元素替换前面的元素, i 就是要返回的长度,为什么i不用加1呢,因为在最后一次执行逻辑中,已经执行了i++
public static 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;
}