给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/remove-element
1.暴力求解
原数组长度为n,根据题设:我们不需要输出所有元素,只要把去除了与val值相等的数组部分的长度输出即可
class Solution
{
public:
int removeElement(vector<int>& nums, int val)
{
int n=size(nums);
for(int i=0;i<n;i++)
{
//当nums中的元素与val相等时,该元素后的所有元素下标前进一位;
if(nums[i]==val)
{
for(int j=i+1;j<n;j++)
{
nums[j-1]=nums[j];
}
i--;//当数组下标偏移后,for循环需要从原本i+1的元素继续遍历,防止遗漏
n--;//每移除一个与val相等的元素,有效的数组长度就-1;
}
}
return n;//返回进行偏移后的有效数组长度即可
}
};
由于使用了嵌套for循环,所以该方法时间复杂度为O();
2.使用快慢指针
定义两个指针:fast;slow;
fast指针进行遍历操作;slow为有效数组长度
当fast遍历时,若元素nums[fast]与val值相等;则fast继续向下遍历;
若不相等,则将自身值传给slow;
最后slow的长度即为有效数组长度。
class Solution
{
public:
int removeElement(vector<int>& nums, int val)
{
int n=size(nums);
//定义快慢指针
int fast=0;
int slow=0;
for(int i=0;i<n;i++)
{
//当nums[fast]==val时,slow指针不变化,fast指针继续向下遍历
if(nums[fast]==val)
{
fast++;
}
else
//当nums[fast]!=val时,fast指针将指向赋给slow;之后两指针同时向后遍历
{
nums[slow]=nums[fast];
fast++;
slow++;
}
}
return slow; //返回的slow的长度即为有效数组长度
}
};
该方法使用了一个for循环,所以此方法的时间复杂度为O(n);