LeetCode 27 ----- 移除元素
题目描述
- 给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
- 不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
- 元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
思路
快慢指针(i 是慢指针,j 是快指针)
nums[j] = val 时,跳过
nums[j] != val 时,nums[i] = nums[j],并且 i++
直至 j 遍历结束,返回 i 的值
时间复杂度 O(n) : i 和 j 至少遍历 2n 步
空间复杂度 O(1) : 没有使用额外的数组空间
实现
/**
* Author: lisiyu
* Created: 2020/1/19
*/
public class RemoveElement27 {
public int removeElement(int[] nums, int val) {
int i = 0;
for (int j = 0; j < nums.length; j++) {
if (nums[j] != val) {
nums[i] = nums[j];
i++;
}
}
return i;
}
}
思路2
为了减少不必要的复制,可以在 nums[i] == val 时,将 nums[i] 与最后一个元素交换,并删除最后一个元素,此时数组长度减小1。
当 nums[i] != val 时,继续遍历。
时间复杂度O(n)
空间复杂度O(1)
实现
/**
* Author: lisiyu
* Created: 2020/1/19
*/
public int removeElement2(int[] nums, int val) {
int i = 0;
int n = nums.length;
while (i < n) {
if (nums[i] == val) {
nums[i] = nums[n - 1];
n--;
} else {
i++;
}
}
return n;
}