给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素。元素的顺序可能发生改变。然后返回 nums
中与 val
不同的元素的数量。
假设 nums
中不等于 val
的元素数量为 k
,要通过此题,您需要执行以下操作:
- 更改
nums
数组,使nums
的前k
个元素包含不等于val
的元素。nums
的其余元素和nums
的大小并不重要。 - 返回
k
。
法一:双重循环暴力解:把所有值为val的数都换到数组最后面去。
void swap(int &x,int &y)
{
int temp=x;
x=y;
y=temp;
}
int removeElement(vector<int>& nums, int val) {//复杂度O(n*n)
int n=nums.size();
int i,j;
for(i=0;i<n;i++){
if(nums[i]==val){//当找到一个val时,向后找一个不为val的数,交换二者
for(j=i;j<n;j++){
if(nums[j]!=val){
swap(nums[i],nums[j]);
break;
}
}
}
}
for(i=0;i<n;i++){//交换完后新数组第一个val的下标即为所求的k
if(nums[i]==val)break;
}
return i;
}
法二:双指针:定义快慢指针,两个指针初始时都指向数组第一个元素,每当数组元素等于一次val时,fast右移一步,slow不动;当数组元不是val时,fast和slow都右移,且fast所指元素向前覆盖slow所指元素(因为没出现val时两个指针指的下标相同,若出现两个指针错位,即说明出现了val值,且第一次覆盖的时候,slow一定指向val值元素,即值为val的数组元素被覆盖)。最终fast领先slow的步数的值即为val元素出现的次数,因此当fast指向数组末尾时,slow所指数组元素下标即为所求k。并且此时slow指针所指下标前面的所有数组元素值均非val。
int removeElement(vector<int>& nums, int val){//复杂度O(n)
int n=nums.size();
int slow=0;
for(int fast=0;fast<n;fast++){
if(nums[fast]!=val){//每相同一次slow就落后fast一步
nums[slow++]=nums[fast];//fast所指的元素往前覆盖
}
}
return slow;//结束时slow下标以前的元素都不等于val
}