题目:
给你一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
思路:
首先我们要明白数组在内存中是一块连续的空间,数组中的元素实际是只能覆盖不能删除,我们要做的就是将目标值所在的位置给覆盖掉,返回一个新长度的数组,而超过此新数组长度之后的元素以及空间我们不做考虑
- 暴力解法
使用两层for循环,第一层循环寻找目标值,第二层循环将目标值之后的元素全部逐一向前移动一位。时间复杂度:O(n^2)
- 双指针
使用快慢指针,已知要从旧数组中寻找目标值并做“删除”操作,我们只需要使用快指针去寻找新数组包含的元素,然后用慢指针指向新数组的当前索引,即快指针找到元素后需要放的位置,这样我们只需要遍历一次数组,即可完成。时间复杂度:O(n)
代码实现:
- 暴力解法
static int deleteNum1(int[] nums, int val) {
int size = nums.length;
for (int i = 0; i < size; i++) {
if (nums[i] == val) {
for (int j = i + 1; j < size; j++) {
nums[j - 1] = nums[j];
}
size--;
}
}
return size;
}
- 双指针解法
static int deleteNum(int[] arr, int target) {
// 慢指针,表示新数组当前索引
int slowIndex = 0;
for (int fastIndex = 0; fastIndex < arr.length; fastIndex++) {
// 快指针指向要放在新数组中的元素索引(不等于target的元素)
if (arr[fastIndex] != target) {
// 将元素放在新数组指定位置后,慢指针加1
arr[slowIndex++] = arr[fastIndex];
}
}
return slowIndex;
}