「每日一题」力扣 27 移动元素

你好啊，我是蓝莓，今天是每日一题的第 2 天。

题目清单：https://blueberry-universe.cn/lc/index.html

引用力扣题目：https://leetcode.cn/problems/remove-element/description/

题目描述

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

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

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

示例 1：

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2,_,_]
解释：你的函数函数应该返回 k = 2, 并且 nums 中的前两个元素均为 2。
你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。

示例 2：

输入：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 个元素之外留下了什么并不重要（因此它们并不计入评测）。

提示：

• 0 <= nums.length <= 100
• 0 <= nums[i] <= 50
• 0 <= val <= 100

实现

思路

这个题目和 「每日一题」力扣 283 移动零 (opens new window) 使用的方法是类似的，我们都是维护一个一开始为空的区间，然后在不断的扫描数组同时维护好我们定义的那个最终答案的区间。

• 维护 nums 数组的区间 [0...k) 为不等于 val 元素的区间

• 由于 区间 [0...k) 装的是不等于 val 的值，所以在算法开始前，我们应该让 k 的值为 0 这样就表示我们在算法开始之前，这个区间内不包含任何元素，每当我们发现一个不等于 val 元素的值后就把这个值和索引 k 所在的值进行交换并让 k 向右移动一个位置

• 在代码实现中有一点是判断 i 和 k 是否相同，如果 i 等于 k 的话，那么我们就没有必要让 nums[i] 和 nums[k] 的值进行交换了，我们只需要让 k 向右移动一个位置就可以让这个元素放进我们维护的区间里面了

Python 代码实现

class Solution(object):
    def removeElement(self, nums, val):
        """
        :type nums: List[int]
        :type val: int
        :rtype: int
        """
        # 维护区间 [0...k) 为非 val 元素的区间
        # 初始化 k 为 0 则 [0...0) 代表为一个空区间
        k = 0

        # 遍历 nums 列表
        for i in range(len(nums)):
            if nums[i] != val:
                if i != k:
                    nums[k], nums[i] = nums[i], nums[k]
                k += 1

        return k

C++ 代码实现

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        // [0...k) 区间维护不等于 val 的元素
        int k = 0;

        for( int i = 0 ; i < nums.size() ; i++ ) {
            if( nums[i] != val ) {
                if( k != i ) {
                    swap(nums[k], nums[i]);
                } 
                k++;
            }
        }

        return k;
    }
};

Java 代码实现

class Solution {
    public int removeElement(int[] nums, int val) {
        int k = 0;

        for( int i = 0 ; i < nums.length ; i++ ) {
            if( nums[i] != val ) {
                if( i != k ) {
                    int temp = nums[i];
                    nums[i] = nums[k];
                    nums[k] = temp;
                }
                k++;
            }
        }

        return k;
    }
}