题目:
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的
样例:示例 1:
cpp
输入: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],也会被视作正确答案。
思路
使用变量k记录等于val的个数,将后面的i元素向前移,移动k个位置,也就是i-k的位置
代码实现
cpp
class Solution {
public:
int removremoveElementeElement(vector<int>& nums, int val) {
int k=0;//用来记录等于val的数组中的元素的个数
int i=0;//遍历
while(i<=nums.size()-1){
if(numsnums[i]!=valval){//当元素不是val的值时执行
nums[i-k]=nums[i];//向前移动数据,补空数据同时使得位序指向下一个元素。
i++;
}else{
//如果恰好遍历到该值,就记录一次k,并且使得i++指向下一个元素。
k++;
i++;
}
}
return nums.size()-k;//总长度减去去除重复元素后的
}
};