题目
给你一个数组 nums
和一个值 val
,你需要原地移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并原地修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
C++代码
#include <iostream>
#include <vector>
using namespace std;
//定义移除数组元素函数
int removeElement(vector<int>& nums, int val) {
int len = 0;
for (int i = 0; i < nums.size(); i++) {
if (nums[i] != val) {
nums[len++] = nums[i];
}
}
return len;
}
int main() {
vector<int> nums = {3,2,2,3};
int val = 3;
int len = removeElement(nums, val);
cout << "len: " << len << endl;
cout << "Removed array: ";
for (int i = 0; i < len; i++) {
cout << nums[i] << " ";
}
}
分析
定义移除数组元素函数 removeElement,在循环条件内判断 nums 中的元素是否与 val 相等,不相等则放入 nums[len++]中,实现元素移除的效果。
问题
发现自己没有搞清楚 while 和 if 的区别,while(){} 是循环语句,会在判断条件内循环判断,直到结果为 false ,if(){} 是条件判断语句,运行完毕后结束。