【数组】-Lc27-移除元素（相向双指针）

写在前面

最近想复习一下数据结构与算法相关的内容，找一些题来做一做。如有更好思路，欢迎指正。

一、场景描述

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

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

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

示例 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,4,0,3]
解释：函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

提示：
0 <= nums.length <= 100
0 <= nums[i] <= 50
0 <= val <= 100

二、具体步骤

1.环境说明

名称	说明
IntelliJ IDEA	2019.2

2.代码

以下为Java版本实现：

java 复制代码

public class Lc27_removeElement {

    public static void main(String[] args) {
        int[] nums = {0, 1, 2, 2, 3, 0, 4, 2};
        System.out.println(removeElement(nums, 2));
//        System.out.println(Arrays.toString(nums));
    }

    /**
     * 思路：相向双指针
     *
     * 返回值是int
     *
     * 定义2个指针left = 0，right = length-1
     *
     * while循环 left <= right
     * left从左向右找等于val的, while(left <= right && nums[left] != val) left++
     * right从右向左找不等于val的, while(left <=right && num[right] == val) right++
     * 如果left<=right, 交换left=right的值，left++, right--
     *
     * 最后返回left
     */
    public static int removeElement(int[] nums, int val) {
        // 左指针找等于val的，右指针找不等于val的，两两交换，左指针就是长度
        int l = 0, r = nums.length - 1;
        while (l <= r) {
            // 找到左边等于val的
            while (l <= r && nums[l] != val) {
                l++;
            }
            // 找到右边不等val的
            while (l <= r && nums[r] == val) {
                r--;
            }
            // r替换l，迭代下一个元素
            if (l <= r) {
                nums[l++] = nums[r--];
            }
        }

        return l;
    }
}

写在后面

如果本文内容对您有价值或者有启发的话，欢迎点赞、关注、评论和转发。您的反馈和陪伴将促进我们共同进步和成长。