27.移除元素 | Thinker的个人网站

题目描述

给你一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并原地修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

题解

这道题属于简单题，知道双指针的概念就可以轻松过关，双指针一般分为两种形式，一种是快慢指针，另一种是左右指针，接下来就是用这两种形式进行解题的。

解法1

/**
 * @param {number[]} nums
 * @param {number} val
 * @return {number}
 */
var removeElement = function(nums, val) {
  var i = 0;
  for (var j = 0; j < nums.length; j++) {
    if (nums[j] !== val) {
      nums[i] = nums[j];
      i++;
    }
  }
  return i;
};

变量 i 是慢指针，变量 j 是快指针，只有当快指针指向的元素不等于 val 值时，将元素赋值给慢指针指向的元素，同时慢指针加一。

解法2

/**
 * @param {number[]} nums
 * @param {number} val
 * @return {number}
 */
var removeElement = function(nums, val) {
  var left = 0, right = nums.length - 1;
  while (left <= right) {
    if (nums[left] !== val) {
      left++;
    } else {
      nums[left] = nums[right];
      right--;
    }
  }
  return left;
};

变量 left 是左指针，变量 right 是右指针，当左指针指向的元素不等于 val 值时，左指针向右移动一位，否则就将右指针指向的元素赋值给左指针指向的元素，同时右指针向左移动一位。

（完）