思路
- 需要使用双指针,即两个用于记录数组下标的变量;
- 左指针从数组第一个元素下标开始右移,左指针之前的元素都是需要保留的元素;
- 右指针从数组最后一个元素下标开始前移,右指针之后的元素都是移除的元素;
- 在每一次循环中,如果元素需要保留,则左指针后移一位;否则将右指针指向的元素赋值到左指针指向的位置(因为当前左指针的元素是要移除的,右指针的元素是未比较过的,所以右指针指向的元素直接覆盖当前要左指针的位置即可),然后右指针前移一位;
- 最后当左指针与右指针重合时,表示所有元素都已经遍历完,左指针所在的位置则是最后一个元素的下标+1,因此长度直接返回左指针即可。
代码
class Solution {
public int removeElement(int[] nums, int val) {
int right = nums.length;
int left;
for (left = 0; left < right;) {
if (nums[left] != val) {
left++;
} else {
nums[left] = nums[right - 1];
right--;
}
}
return left;
}
}
题目链接
- 移除元素