题目:给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。
注意:元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
思路:这道题我现在考虑的解法有两种:①暴力解法;②双指针。
①暴力解法(不改变元素顺序)
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int size = nums.size(); // 获取数组的长度
for (int i = 0; i < size; i++) { // 遍历数组
if (nums[i] == val) { // 如果当前元素等于给定值 val
for (int j = i + 1; j < size; j++) { // 从当前位置的下一个位置开始遍历数组
nums[j - 1] = nums[j]; // 将后面的元素依次向前移动一位,相当于删除当前值为 val 的元素
}
i--; // 将指针退回到上一个位置,因为在删除元素后,当前位置的元素已经被覆盖为了后面的元素,需要重新检查该位置
size--; // 更新数组的长度,因为删除了一个元素
}
}
return size; // 返回移除元素后数组的新长度
}
};
①暴力解法(改变元素顺序)
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int n = nums.size(); // 数组长度
int i = 0; // 当前遍历到的位置
while (i < n) { // 遍历整个数组
if (nums[i] == val) { // 如果当前元素等于给定值
nums[i] = nums[n - 1]; // 将当前元素与数组末尾元素交换位置
n--; // 更新数组长度,相当于删除当前元素
} else {
i++; // 否则继续向后遍历
}
}
return n; // 返回新数组的长度
}
};
②双指针
双指针技巧通常有三种类型:
-
快慢指针:用于解决链表中的问题,快指针一次移动两步,慢指针一次移动一步,常用于判断链表是否有环或找到链表的中间节点等问题。
-
左右指针:在数组中经常使用,指向数组的两端,根据题目要求,通过移动左右指针来缩小搜索范围,或者解决一些需要同时遍历数组的问题。
-
对撞指针:也是左右指针的一种特例,在有序数组中使用,将指针分别放在数组的两端,根据题目的要求移动左右指针,以寻找特定的目标。
class Solution {
public:
// 移除数组中所有值为给定值 val 的元素,并返回移除后数组的新长度
int removeElement(vector<int>& nums, int val) {
int slow = 0; // 慢指针,表示新数组的下一个插入位置
int n = nums.size(); // 获取数组的大小,即元素个数
// 遍历数组,查找需要移除的元素
for (int fast = 0; fast < n; fast++) {
// 如果当前元素不等于给定值 val
if (nums[fast] != val) {
// 将当前元素复制到新数组的 slow 位置
nums[slow++] = nums[fast];
}
}
// 返回新数组的长度,即慢指针的值
return slow;
}
};
谢谢大家,继续努力。