【数组】-Lc27-移除元素(相向双指针)
写在前面
最近想复习一下数据结构与算法相关的内容,找一些题来做一做。如有更好思路,欢迎指正。
目录
- 写在前面
- 一、场景描述
- 二、具体步骤
- 1.环境说明
- 2.代码
- 写在后面
一、场景描述
给你一个数组 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版本实现:
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;
}
}
写在后面
如果本文内容对您有价值或者有启发的话,欢迎点赞、关注、评论和转发。您的反馈和陪伴将促进我们共同进步和成长。