数据结构 ——— 移除元素(快慢指针)
目录
题目要求
代码实现(快慢指针)
题目要求
编写函数,给你一个数组 nums 和一个值 val,你需要在 nums 数组 原地 移除所有数值等于 val 的元素,并且返回移除后数组的新长度
不能使用额外的数组空间,要保证 算法的空间复杂度在 O(1) 内
元素的顺序可以改变,且不需要考虑数组中超出新长度后面的元素
代码实现(快慢指针)
代码演示:
int removeElement(int* nums, int numsSize, int val)
{
int slow = 0;
// 循环1
for (int fast = 0; fast < numsSize; fast++)
{
if (nums[fast] != val)
{
nums[slow] = nums[fast];
slow++;
}
}
return slow;
}
代码解析:
创建两个 int 类型的变量,slow 和 fast ,用来充当 nums 数组的快慢指针的作用
slow 的作用是从 nums 数组的起始位置开始存储非 val 的值,成功存储了一个非 val 的值后,slow 才会指向下一个位置,也就是 slow 自增 1
fast 的作用是从 nums 数组的起始位置开始找到非 val 的值,找到后就赋值给 nums[slow] ,然后再往后找非 val 的值,直到找完 nums 数组的最后一个元素
代码验证:
算法的时间复杂度:
循环1 执行了 N 次,每次循环内部执行了常数次
得出算法的时间复杂度(大O渐进表示法):O(N)
算法的空间复杂度:
没有开辟额外的临时空间,是在原 nums 数组上操作的
得出算法的空间复杂度(大O渐进表示法):O(1)