【6.移除元素】

12 阅读1分钟

题目

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

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

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

示例 :

输入: nums = [0,1,2,2,3,0,4,2], val = 2
输出: 5, nums = [0,1,4,0,3,_,_,_]
解释: 你的函数应该返回 k = 5,并且 nums 中的前五个元素为 0,0,1,3,4。
注意这五个元素可以任意顺序返回。
你在返回的 k 个元素之外留下了什么并不重要(因此它们并不计入评测)。

题解

方式一:双指针

复杂度:O(n)

// 题目要求nums前k个元素不包含val
public int removeElement(int[] nums, int val) {
    int left = 0;
    for (int right = 0; right < nums.length; right++) {
        if (nums[right] != val) {
            // 要保留的元素赋值给left
            nums[left] = nums[right];
            left++;
        }
    }
    return left;
}

方式二:双指针

复杂度:O(n)

// 因为可以乱序,所以只移动不保留的元素可以减少操作次数
public int removeElement(int[] nums, int val) {
    int left = 0;
    int right = nums.length;
    while (left < right) {
        if (nums[left] == val) {
            // 不保留的赋值给right - 1
            nums[left] = nums[right - 1];
            right--;
        } else {
            left++;
        }
    }
    return left;
}

总结

算法:双指针