leetcode-note/remove-element.js at master · qhx0807/leetcode-note · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
/**
 * 移除元素
 * 须在原地修改输入数组
 * @param {number[]} nums
 * @param {number} val
 * @returns {number}
 * @example
 * nums = [3, 2, 2, 3], val = 3
 * // returns [2, 2]
 */
const removeElement = (nums, val) => {
  if (nums.length === 0) return 0
  /**
   * 1. 原始的方法
   */
  // let arr = []
  // for (let i = 0; i < nums.length; i++) {
  //   if (nums[i] !== val) {
  //     arr.push(nums[i])
  //   }
  // }
  // if (arr.length === 0) return 0
  // for (let j = 0; j < arr.length; j++) {
  //   nums[j] = arr[j]
  // }
  // return arr.length

  /**
   * 2. 方法改进
   * 要从数组尾部开始遍历，才能splice不影响数组
   */
  for (let i = nums.length; i >= 0; i--) {
    if (nums[i] === val) nums.splice(i, 1)
  }
  return nums.length

  /**
   * 3. 双指针法
   * 一个快指针，一个慢指针
   * TODO: 双指针，快慢指针
   */
  // let i = 0
  // for (let j = 0; j < nums.length; j++) {
  //   if (nums[j] !== val) {
  //     nums[i] = nums[j]
  //     i++
  //   }
  // }
  // return i
}

console.log(removeElement([3, 2, 2, 3], 3))