LC27删除元素

LeetCode 问题：删除元素

在这篇博客中，我们将讨论 LeetCode 上一个常见的面试问题：删除元素。这个问题在面试中经常遇到，是一个很好的练习，可以帮助你提高对数组操作的理解。

问题描述

给定一个整数数组 nums 和一个整数 val，你需要原地删除数组中所有等于 val 的元素。元素的顺序可以改变，你应该返回删除指定值后数组的新长度。

示例

输入: nums = [3, 2, 2, 3], val = 3
输出: 2, nums = [2, 2]

输入: nums = [0,1,2,2,3,0,4,2], val = 2
输出: 5, nums = [0,1,3,0,4]

约束条件

• 你必须以 O(1) 的额外空间复杂度解决此问题。
• 函数应在原地修改输入数组。

方法

解决这个问题的关键是使用双指针技术。我们将维护两个指针：一个（left）用于遍历数组，另一个（right）用于跟踪可以放置非 val 元素的位置。

算法步骤

1. 初始化两个指针：

   • left 从数组开头开始。
   • right 从数组末尾开始。

2. 使用 while 循环遍历，直到 left 小于 right。

   • 移动 left 指针，直到找到一个等于 val 的元素。
   • 移动 right 指针，直到找到一个不等于 val 的元素。
   • 如果 left 仍然小于 right，则交换这两个指针指向的元素。

3. 循环结束后，检查 left 指针指向的元素是否等于 val。如果是，返回 left 作为新长度；如果不是，返回 left + 1。

实现

以下是用 Java 实现的解决方案：

class Solution {
    public int removeElement(int[] nums, int val) {
        if(nums.length == 0 || nums == null){
            return 0;
        }
        int left = 0;
        int right = nums.length - 1;
        while(left < right){
            while(left < right && nums[left] != val){
                left++;
            }
            while(left < right && nums[right] == val){
                right--;
            }
            int temp = nums[left];
            nums[left] = nums[right];
            nums[right] = temp;
        }
        if(nums[left] == val){
            return left;
        }else{
            return left + 1;
        }
    }
}

代码解释

1. 首先检查输入数组是否为空或为 null。
2. 初始化 left 和 right 指针。
3. 进入循环，继续处理直到 left 小于 right。
4. 在循环中：
   • 移动 left 向前，直到找到一个等于 val 的元素。
   • 移动 right 向后，直到找到一个不等于 val 的元素。
   • 如果 left 仍然小于 right，则交换这两个指针指向的元素。
5. 最后，检查 left 指针指向的元素以确定新长度。

总结

删除元素 问题是一个很好的练习，可以帮助你熟悉数组操作和双指针技术。通过这种方法，我们可以有效地删除元素，同时保持 O(n) 的时间复杂度和 O(1) 的空间复杂度。