移除元素

2025年3月28日

分析

这题的核心就是快慢指针，思路本身并不复杂。

我们可以取两个指针，慢指针用于逐个读取元素，快指针在 nums[fast] = val 时跳过。然后我们仅需返回 slow 的值。这样一来时间复杂度为 $O(N)$ ，空间复杂度为 $O(1)$ 。

解答

1
class Solution {
2
public:
3
  int removeElement(vector<int> &nums, int val) {
4
    auto slow = 0;
5

6
    for (auto fast = 0; fast < nums.size(); ++fast) {
7
      if (nums[fast] != val)
8
        nums[slow++] = nums[fast];
9
    }
10

11
    return slow;
12
  };
13
};