题目

移除元素 - 图1

解题思路

如果要移除的元素恰好在数组的开头,例如序列 [1,2,3,4,5],当 val 为 1 时,我们需要把每一个元素都左移一位。

注意到题目中说:「元素的顺序可以改变」。实际上我们可以直接将最后一个元素 5 移动到序列开头,取代元素 1,得到序列 [5,2,3,4],同样满足题目要求。这个优化在序列中 val 元素的数量较少时非常有效。

实现方面,我们依然使用双指针,两个指针初始时分别位于数组的首尾,向中间移动遍历该序列。

如果左指针left 指向的元素等于 val,此时将右指针 right 指向的元素复制到左指针 left 的位置,然后右指针 right 左移一位。如果赋值过来的元素恰好也等于 val,可以继续把右指针right 指向的元素的值赋值过来(左指针 left 指向的等于val 的元素的位置继续被覆盖),直到左指针指向的元素的值不等于val 为止。

当左指针left 和右指针 right 重合的时候,左右指针遍历完数组中所有的元素。

这样的方法两个指针在最坏的情况下合起来只遍历了数组一次。这避免了需要保留的元素的重复赋值操作。

代码

  1. class Solution {
  2. public int removeElement(int[] nums, int val) {
  3. int n = nums.length;
  4. int left = 0;
  5. for (int right = 0; right < n; right++) {
  6. if (nums[right] != val) {
  7. nums[left] = nums[right];
  8. left++;
  9. }
  10. }
  11. return left;
  12. }
  13. }