题目描述
给你一个数组 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 <= 10
- 0 <= nums[i] <= 50
- 0 <= val <= 100
我的解法
思路
题干中要求不要使用额外的数组空间,那就是要对原数组进行改动;
从示例可以看出,从数组中删掉等于val的元素,或者把它们全面全部移到最后都是可以的,不过真正删掉相当于对原数组空间进行了变动,相比之下进行移动可能是更好的选择。
在对数组进行遍历时同时修改数组这种行为很容易出现错误,这是难点所在
能直接想到的解法是从前面开始遍历,碰到等于val的元素将其与后面不等于val的元素互换,前后各设置一个标识点代表遍历的位置,虽然整个数组只是遍历了一遍,但依然是两层循环
对应Java代码
class Solution {
public int removeElement(int[] nums, int val) {
int j = nums.length - 1;
int num = 0;
for (int i = 0; i <= j; i++) {
if (nums[i] == val) {
while (nums[j] == val && i != j) {
j--;
num++;
}
num++;
int t = nums[j];
nums[j] = 0;
nums[i] = t;
j--;
}
}
return nums.length - num;
}
}
【过程回顾】判断条件的边界考虑的还是不清晰,提交过程中出了很多次错误,对数组进行遍历的时候应该设定个条件涉及到上述代码中 i<=j 是否有等于
,比如这里面就要求数组中的每个元素都要被遍历到,并且要考虑边界值时的处理条件与一般情况是否一致涉及到whie的条件中是否需要添加 i != j 这条
。
复杂度分析
时间复杂度:O(n²)
空间复杂度:O(1)
更优解法
思路
此部分转载于公众号代码随想录
双指针法(快慢指针法):
「通过一个快指针和慢指针在一个for循环下完成两个for循环的工作。」
删除过程如下:
「双指针法(快慢指针法)在数组和链表的操作中是非常常见的,很多考察数组和链表操作的面试题,都使用双指针法。
其他用到的场景还有以下:
- 15.三数之和
- 18.四数之和
双指针来记录前后指针实现链表反转:
- 206.反转链表
使用双指针来确定有环:
- 142.环形链表Ⅱ
对应Java代码
class Solution {
public int removeElement(int[] nums,int val) {
int slowIndex = 0;
for (int fastIndex = 0; fastIndex < nums.length ; fastIndex++) {
if (nums[fastIndex] != val) {
nums[slowIndex++] = nums[fastIndex];
}
}
return slowIndex;
}
}
复杂度分析
时间复杂度:O(n)
空间复杂度:O(1)
收获总结
学习到了可以将两个for循环变成一个for循环的快慢指针法,但是只从这一道题理解还是不到位,接下来把上面几道相关的题目都做一下