当前位置：首页 > article >正文

leetcode 之（移除元素）

article 2025/3/27 18:18:21

在这里插入图片描述
给你一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。

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

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

思路

使用左右双指针,左指针从左侧开始找需要移除的元素，右指针从右往左找不需要移动的元素，找到后，把右指针的值赋给当前左指针，然后继续往中间找：

左右侧指针重合时：判断当前值是否需要移除需要返回指针索引值，不需要返回指针索引值+1；
左指针在右指针右侧时：返回左指针索引值；

class Solution {
    public int removeElement(int[] nums, int val) {
        // 定义等于val 时的默认值
        // 双指针 前后指针
        int left = 0, right = nums.length - 1;
        while (left <= right) {
            // 如果左指针等于val 右指针不等于val
            if (val == nums[left] && val != nums[right]) {
                // 把右指针的值赋给左指针 移动左指针
                nums[left++] = nums[right--];
            } else if (val != nums[left]) {
                // 如果左指针不等于val 移动左指针
                left++;
            }
            // 如果右指针等于val，移动右指针
            if (val == nums[right]) {
                right--;
            }
            // 如果左指针位置不在右指针左侧，
            if (left >= right) {
                // 1.左指针位置在右指针左侧，说明已经全部移动完成，返回左指针位置（不包含当前位置）
                // 2.如果左右指针在同一位置 判断 当前位置的值是否等于val
                // 2.1 如果等于val 返回左指针位置（不包含当前位置）
                // 2.2 如果不等于val 返回 左指针位置+1（包含当前位置）
                return left > right || val == nums[left] ? left : ++left;
            }
        }
        return 0;
    }
}