题目描述:
给你一个数组 nums 和一个值 val,你需要“原地”移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
题目其实想的就是给出移除后的个数,然后使用个数遍历给的数组
所以一般的思路就是使用双指针将等于val的值依次移到数组末尾
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int left=0,right=nums.size()-1;
int res=0;
while(1){
if(left>right)return left;
if(nums[left]!=val){
left++;
res++;
}else if(nums[right]==val){
right--;
}else{
nums[left]=nums[right];
nums[right]=val;
left++;
right--;
}
}
return 0;
}
};
但我想的思路就是删除就删除呗,使用vector容器的删除函数直接删除就是了,其实也是很好理解(当然也顺便学习了容器的用法)
注意:it=nums.erase(it)就是vector容器的删除指定元素的函数,返回的是删除之后它将返回下一个位置的迭代器
所以当删除时其实就相当于更新了,所以要分情况
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int n=0;
for(vector<int>::iterator it=nums.begin();it!=nums.end();){
if(*it==val)it=nums.erase(it);
else{
it++;
n++;
}
if(it==nums.end())break;
}
return n;
}
};