移除元素(双指针实现)
题目
给你一个数组 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;
}
下一篇:
实现背景跟随图片变化
