双指针应用之移除元素
移除元素
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++;
}
}