leetcode 27 移除元素

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。

假设 nums 中不等于 val 的元素数量为 k，要通过此题，您需要执行以下操作：

  • 更改 nums 数组，使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
  • 返回 k。

思路

快慢指针法

设置 slow, fast两个指针，初始均为0

  • fast用于遍历数组
  • slow用于记录当前保留的元素

无效元素：fast后移，slow不移；

有效元素：放入 slow指针位置，fast、slow均后移

1
2
3
4
5
6
7
8
9
10
11
12
class Solution {
    public int removeElement(int[] nums, int val) {
        int slow=0;
        for(int fast=0;fast<nums.length;++fast){
            if(nums[fast]!=val){
                nums[slow]=nums[fast];
                ++slow;  
            }
        }
        return slow;
    }
}
  • 时间复杂度：O(n)
  • 空间复杂度：O(1)