算法原理
多余的元素,删掉不就得了。
但是数组的元素在内存地址中是连续的,不能单独删除数组中的某个元素,只能覆盖。
这个题目暴力的解法就是两层for循环,一个for循环遍历数组元素 ,第二个for循环更新数组。
// 时间复杂度:O(n^2)
// 空间复杂度: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) { // 发现需要移除的元素,就将数组集体向前移动一位
for (int j = i + 1; j < size; j++) {
nums[j - 1] = nums[j];
}
i--; // 因为下标i以后的数值都向前移动了一位,所以i也向前移动一位
size--; // 此时数组的大小-1
}
}
return size;
}
};
使用双指针法(快慢指针法),可以节省时间复杂度。通过一个快指针和慢指针在一个for循环下完成两个for循环的工作。
定义快慢指针
- 快指针:寻找新数组的元素 ,新数组就是不含有目标元素的数组
- 慢指针:指向更新 新数组下标的位置
两个指针都指向一个数组,fast的任务是找到符合要求的新数组元素,slow则对应 新数组元素的下标。
代码调试过程
列表名是nums,调用元素时要用nums[slow],中括号!!!
修改后的代码:
原因:return slow在while循环里面,当第一次while循环时,3=3,所以slow=0,return回空值。由于return slow在while循环里面,return一次以后就打断了循环。
当在 while 循环内部执行 `return
` 语句时,会立即中断循环并返回相应的值。
所以,`return slow
`语句应该位于循环的外部,以便在遍历完整个列表后返回结果。
继续修改:
原因:fast指针+1的位置不对。如果 `nums[fast]
` 不等于指定的值 `val
`,会执行 `nums[slow] = nums[fast]
`、`slow += 1
` 和 `fast += 1
`,这会导致 `slow
` 和 `fast
` 同时递增。这样会导致处理元素的跳过,因为下一个元素将被放置到 `nums[slow]
` 的位置,但 `slow
` 的值已经递增了。这将导致结果列表中的元素数量少于预期,或者在处理元素时出现错误。
-
`
slow += 1
` 在 `if nums[fast] != val
` 的条件下递增。这是因为只有当 `nums[fast]
` 不等于指定的值 `val
` 时,才需要将该元素移动到 `nums
` 的 `slow
` 索引位置。因此,只有在这个条件下,我们才会递增 `slow
` 的值,以确保下一个非指定元素能够被正确地放置。 -
`
fast += 1
` 则是在每次迭代中都递增。它的目的是在遍历列表时逐个检查元素,并将 `fast
` 递增以处理下一个元素。无论当前元素是否等于指定的值 `val
`,都需要递增 `fast
`,以便在下一次循环中检查下一个元素。
正确的代码
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
slow=0
fast=0
size=len(nums)
while fast<size:
if nums[fast]!=val:
nums[slow]=nums[fast]
slow+=1
fast+=1
return slow
使用enumerate函数,它允许你在遍历列表或其他可迭代对象时,同时获取元素的索引和值:
#双指针法
#快指针:找目标元素
#慢指针:更新index
#enumerate函数,它允许你在遍历列表或其他可迭代对象时,同时获取元素的索引和值
# nums = [10, 20, 30, 40, 50]
# # 使用 enumerate 获取索引和值
# for index, value in enumerate(nums):
# print(f"Index: {index}, Value: {value}")
# Index: 0, Value: 10
# Index: 1, Value: 20
# Index: 2, Value: 30
# Index: 3, Value: 40
# Index: 4, Value: 50
#本质是双指针法:通过一个简单的循环和双指针技巧正确地移除了数组中的指定元素。
class Solution(object):
def remove(self, nums, val):
new = 0
for r, x in enumerate(nums):
if (x != val):
nums[new] = x
new += 1
return new
2024.8.26三刷
class Solution(object):
def removeElement(self, nums, val):
new = 0
for x in nums:
if (x != val):
nums[new] = x
new += 1
return new
其他
其实这个代码也可以直接调用remove、erase等内嵌函数,但是这样只要一句话就能实现了,显然不是面试所需。
什么时候可以用内嵌函数?
当该函数只是解题的一个步骤,并且我们知道该函数的时间、空间复杂度。