穿越算法迷宫：LeetCode 27 巧妙去除指定元素

2023-10-21 04:42:45







## 算法探索：巧夺数组指定元素

想象一个场景，你手握一个数字数组，其中某些元素需要被无情地去除。LeetCode 27 题正是这样一个场景，它要求你从给定的数组中原地删除所有指定元素。

面对这样的任务，我们自然想到遍历数组，逐一检查元素是否与目标元素相等。然而，这样的做法存在效率问题，因为每次删除元素都需要移动数组中的所有元素，时间复杂度为 O(n^2)。显然，这并不是一个理想的解决方案。

为了提高算法效率，我们引入了双指针法。双指针法通过两个指针同时遍历数组，一个指针指向当前元素，另一个指针指向数组中第一个未被访问的元素。当当前元素不与目标元素相等时，我们移动未访问指针，否则移动当前指针，最终完成所有元素的遍历。

具体来说，我们从数组的开头和结尾各设置一个指针。如果当前指针指向的元素不与目标元素相等，则移动未访问指针。如果相等，则交换当前指针和未访问指针指向的元素，移动当前指针，同时保持未访问指针不变。这样一来，未访问指针始终指向第一个未被访问的元素，而当前指针指向最后一个与目标元素相等的元素。

当未访问指针超过数组的结尾时，遍历结束，数组中所有指定元素已从左至右排列。我们只需截取数组中未被访问指针之前的元素，即可得到去除指定元素后的新数组。

```java
    public int removeElement(int[] nums, int val) {
        if (nums == null || nums.length == 0) {
            return 0;
        }

        int i = 0;
        int j = nums.length - 1;
        while (i <= j) {
            if (nums[i] == val) {
                nums[i] = nums[j];
                j--;
            } else {
                i++;
            }
        }

        return j + 1;
    }