题目描述:
给你一个数组 nums 和一个值 val,你需要“原地”移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
题目连接-力扣27. 移除元素
题目其实想的就是给出移除后的个数,然后使用个数遍历给的数组
所以一般的思路就是使用双指针将等于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; } };