LeetCode题练习与总结：移除元素

一、题目描述

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

说明:

为什么返回数值是整数，但输出的答案是数组呢?

请注意，输入数组是以「引用」方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下:

// nums 是以“引用”方式传递的。也就是说，不对实参作任何拷贝
int len = removeElement(nums, val);

// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}

示例 1：

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2]
解释：函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如，函数返回的新长度为 2 ，而 nums = [2,2,3,3] 或 nums = [2,2,0,0]，也会被视作正确答案。

示例 2：

输入：nums = [0,1,2,2,3,0,4,2], val = 2
输出：5, nums = [0,1,3,0,4]
解释：函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

提示：

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

二、解题思路

1. 使用双指针技术。一个指针用于遍历数组（我们称之为 i），另一个指针用于标记新数组的末尾（我们称之为 j）。
2. 遍历数组 nums，检查每个元素是否等于 val。
3. 如果当前元素不等于 val，将其移动到 j 指针指向的位置，然后递增 j。
4. 如果当前元素等于 val，直接递增 i，不做任何操作。
5. 遍历完成后，j 的值即为新数组的长度。

三、具体代码

class Solution {
    public int removeElement(int[] nums, int val) {
        if (nums == null || nums.length == 0) {
            return 0;
        }
        int j = 0; // 新数组的末尾位置
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] != val) {
                nums[j] = nums[i];
                j++;
            }
        }
        return j; // 返回新数组的长度
    }
}

四、时间复杂度和空间复杂度

1. 时间复杂度

• 时间复杂度：O(n)。
• 该算法遍历了整个数组一次，对于数组中的每个元素，我们进行了常数时间的操作（比较和可能的赋值）。
• 因此，时间复杂度是 O(n)，其中 n 是数组 `nums` 的长度。

2. 空间复杂度

• 空间复杂度：O(1)。
• 算法没有使用额外的数组来存储数据，所有的操作都是在原数组上进行的。
• 我们只使用了两个额外的变量 `i` 和 `j`，它们的空间需求是常数的，不依赖于输入数组的大小。
• 因此，空间复杂度是 O(1)，即常数空间复杂度。

五、总结知识点

1. 数组处理：代码处理了一个整型数组 nums，并在数组上进行了原地修改。

2. 双指针技术：这是一种常见的算法技巧，用于在遍历数组的同时进行修改。在这里，i 指针用于遍历数组，而 j 指针用于标记新数组的末尾位置。

3. 原地操作：代码没有使用额外的数组来存储结果，而是直接在原数组上进行修改，这符合题目要求的 O(1) 额外空间复杂度。

4. 条件判断：在遍历过程中，使用 if 语句来判断当前元素是否等于要移除的值 val。

5. 数组元素赋值：当当前元素不等于 val 时，将其赋值给 j 指针指向的位置，并将 j 递增。

6. 边界检查：在进行数组操作之前，代码首先检查数组是否为空或长度为0，这是一个良好的编程习惯，可以避免潜在的空指针异常。

7. 空值检查：在处理数组之前，代码检查了数组是否为 null，这是处理可能的空指针异常的一种方式。

以上就是解决这个问题的详细步骤，希望能够为各位提供启发和帮助。