给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
输入:nums = [3,2,2,3], val = 3 输出:2, nums = [2,2] 解释:函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如,函数返回的新长度为 2 ,而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也会被视作正确答案。
题解:
直接从结果开始思考,移除后,前面部分是我们需要输出的不含val的元素,那么,我们用一个变量来记住出现过多少次val,并且把所有的val丢在后面即可。
那么还是逆向遍历,用n来记住目前出现过几次val,从后往前遍历,到遇到val的时候(n先不加当前遇见的这个)(意味着,后面有n个都是不要的,从当前遍历到的位置到len-n-1的这部分是遍历过并且能够保证不是val的),我们把nums[len-n-1]的值放到当前位置即可(由于无需输出全部内容,所以不必交换,保证前面是好的就可以)。
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int n=0;
int temp;
int len=nums.size();
for(int i=len-1;i>=0;i--){
if (nums[i]==val){
//temp=nums[i];//无需考虑后面的,所以不用交换,修改前面部分即可
nums[i]=nums[len-n-1];
//nums[len-n-1]=temp;
n++;
}
}
return len-n;
}
};