给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
给定 nums = [3,2,2,3], val = 3, 函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。 你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以“引用”方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参作任何拷贝 int len = removeElement(nums, val); // 在函数里修改输入数组对于调用者是可见的。 // 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。 for (int i = 0; i < len; i++) { print(nums[i]); }
第一个段是我自己敲的代码,遇到重复的就直接将当前位置后面的数都往前移动一位,检查的时候也就少遍历一位。因为不考虑顺序,所以这种方法有点儿麻烦。因此贴上第二段代码。
#include <iostream>
#include <string>
#include <vector>
using namespace std;
int removeElement(vector<int>& nums, int val)
{
int count = 0;
for (int i = 0;i < nums.size() - count;i++)
{
if (nums[i] == val)
{
count++;
for (int j = i;j < nums.size() - 1;j++)
{
nums[j] = nums[j + 1];
}
i--;
}
}
for (int i = 0;i < nums.size();i++)
{
cout << nums[i] << " ";
}
cout << endl;
return count;
}
int main()
{
int arr[] = { 0,1,2,2,3,0,4,2 };
vector<int> nums(arr,arr + 8);
cout << removeElement(nums, 2) << endl;
return 0;
}
第二段代码的核心思想是将第i个与val值相等的数与第n位交换,然后n--,那么nums[i]就变成nums[n]消失了。若不相等i++即可。
public class Solution
{
public int removeElement(int[] nums, int val)
{
int n=nums.length;
int i=0;
while(i<n)
{
if(nums[i]==val)
{
n--;
nums[i]=nums[n];
}
else
{
i++;
}
}
return n;
}