双指针应用之移除元素

双指针应用之移除元素

移除元素

https://leetcode-cn.com/problems/remove-element/

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

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

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

示例 1:

给定 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。

注意这五个元素可为任意顺序。

你不需要考虑数组中超出新长度后面的元素。

分析：

解法1） 双指针覆盖

[3,2,1,3] val = 3

使用额外指针 记录要覆盖的索引位置
索引
3 0 [3,2,1,3]
2 1 [2,2,1,3]
1 2 [2,1,1,3]
3

[0,1,2,2,3,0,4,2] val = 2
索引初始值是0
元素 索引 数组
0 1 [0,1,2,2,3,0,4,2]
1 2 [0,1,2,2,3,0,4,2]
2
2
3 3 [0,1,3,2,3,0,4,2]
0 4 [0,1,3,0,3,0,4,2]
4 5 [0,1,3,0,4,0,4,2]
2

public static int removeElement(int[] nums, int val) {
        //  记录要覆盖的索引位置
        int index = 0;

        for (int i = 0; i < nums.length; i++) {
            if (nums[i] != val) {
                // 覆盖的是当前位置
                nums[index] = nums[i];
                System.out.println("将" + nums[i] + "覆盖到" + index + "位置上");
                index++;
            }

        }
        // 新数组的长度大小
        return index;
    }

解法2）直接替换

[0,1,2,2,3,0,4,2] val=2

[0,1,2,2,3,0,4|,2] len=7

[0,1,4,2,3,0|,2,2] len=6

[0,1,4,0,3|,2,2,2] len=5

index=0 len=8

判断当前值 是否需要被删除
如果需要 进行替换 替换最后一个位置的元素
nums[index] = nums[len-1]
len–;
index–; // 替换回来的值 仍要进行判断

index++

当index >= len 循环结束

   public static int removeElement1(int[] nums, int val) {
        int index = 0;
        // 记录有效元素的个数
        int len = nums.length;

        while (true) {
            if (index >= len) return len;

            if (nums[index] == val) {
                nums[index] = nums[len - 1];
                nums[len - 1] = val;

                System.out.println("替换位置" + index + "和" + (len - 1) + "的元素");
                System.out.println(Arrays.toString(nums));

                len--;
                index--;
            }
            index++;
        }
    }