移除元素
本来想着每天一题,做完一看下一题,这个我熟,啪一下就写出来了,很快啊!因为和刚刚做的题目实在是太像了:删除排序数组中的重复项,返回数组长度
一、题目:
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
二、解题思路:
使用一个循环遍历数组,
如果元素不等于val,那么就把nums[i]复制给nums[j],j++;
返回j;
三、代码和结果:
方法一:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int j = 0;
for(int i=0; i<nums.size(); i++){
if(nums[i] != val){//只要每个元素不等于val,就把nums[i]加入到nums[j],
nums[j] = nums[i];
j++;
}
//如果nums[i] = val,nums[i]就不加入nums[j]
}
return j;
}
};
结果:
方法二:(PS:似乎并不满足题目中 你必须仅使用 O(1) 额外空间并 原地 修改输入数组)
前两天刚好看了一下STL的list,他自带直接移除某个元素的函数,所以;
我们可以利用STL列表中可以直接移除某个元素的特性完成本题,思路都在代码里了:
如下:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
//定义一个列表
list<int>l;
//将数组中的元素存入列表中
for(auto m:nums){
l.push_back(m);
}
//移除列表中所有值为val的元素
l.remove(val);
//清空数组
nums.clear();
//列表值放入数组中去
for(auto m:l){
nums.push_back(m);
}
//返回列表(数组)大小
return l.size();
}
};
怕什么真理无穷,进一步有进一步的欢喜!