移除元素(双指针实现)
题目
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
注意: 1、不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并原地修改输入数组; 2、元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 : 给定 nums = [3,2,2,3], val = 3, 函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。 你不需要考虑数组中超出新长度后面的元素。
注意点
1、指针“i”记录需要遍历数组的当前位置,指针“end”记录需要遍历数组的最后一个元素的位置; 2、当前元素与val值不同,则下标往后移动;当前元素与val值相同,最后面元素替换当前位置(相当于删除当前元素(val),并将最后的元素移动到当前位置); 3、循环结束条件为i <= end,返回结果为i。
实现
public int removeElement(int[] nums, int val) { if(nums.length == 0) return 0; //记录需要遍历数组的当前位置 int i = 0; //记录需要遍历数组的最后一个元素的位置 int end = nums.length - 1; while(i <= end){ if(nums[i] != val){ //当前元素与val值不同,则下标往后移动 i ++; }else{ //当前元素与val值相同,最后面元素替换当前位置 nums[i] = nums[end --]; } } return i; }
下一篇:
实现背景跟随图片变化