题目描述
给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 nums = [0,1,2,2,3,0,4,2], val = 2, 函数应该返回新的长度5
, 并且 nums 中的前五个元素为0
,1
,3
,0
, 4。 注意这五个元素可为任意顺序。 你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以“引用”方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参作任何拷贝
int len = removeElement(nums, val);
// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。
for (int i = 0; i < len; i++) {
print(nums[i]);
}
解决方案
我就直接上代码吧,不知道怎么回事,Leetcode竟然说我的代码击败了100.00%的用户(注意博客发布时间,有可能后来人有更好的方法,这样就不是击败100%了。哈哈,所以别喷我哈!),不可思议。题目虽然简单,但是里面C++中STL中的某些方法还是很有必要好好学习的,学好了并掌握了这些STL方法,在以后的工作必能事半功倍。先看代码。然后说明一下原理和方法细节。
好了,不装ACD了,下面就看代码吧!
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
auto ibegin=nums.begin();
auto iend=nums.end();
while(ibegin!=iend)
{
if(*ibegin==val)
{
ibegin=nums.erase(ibegin);
iend=nums.end();//一定要注意,当删除了一个元素之后,这个vector的元素已经向前移动一个元素,但是呢end()指针发生了变化。
}
else
ibegin++;
}
return nums.size();
}
};
原理以及方法
为了更好的移除指定内容的vector元素,肯定需要进行一次迭代,所以我定义两个迭代器。
但是应该注意erase的使用方法,为了做这个题,我甚至还查看了官方vector的使用说明(见下图):
可以看出他有两个使用方法:
- 给一个迭代器,让他删除指定元素
- 给一个迭代器的范围,他会删除范围内的所有元素。
可以看出来,需要指出删除的迭代器指针,然后返回值是当前所删除元素的下一个位置。所以呢一旦你进行了这个操作,就不要再自己ibegin++了,直接将返回值赋给ibegin就可以了。(我们使用1的方法。)
ibegin=nums.erase(ibegin);
还有一点需要注意:
iend=nums.end();//一定要注意,当删除了一个元素之后,这个vector的元素已经向前移动一个元素,但是呢end()指针发生了变化。
因为删除了vector中的元素,这时候最后一个元素就移动了,这样iend所指的东西就不知道是什么地方了,因此需要实时更新最后一个指针(别看这个问题很小,花我好多时间调试。)所以凡是涉及到容器增删的时候,迭代器范围指针,务必要进行实时更新。
这就是我的总结,希望大家多多关注,多多交流。