给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素。元素的顺序可能发生改变。然后返回 nums
中与 val
不同的元素的数量。
假设 nums
中不等于 val
的元素数量为 k
,要通过此题,您需要执行以下操作:
更改 nums
数组,使 nums
的前 k
个元素包含不等于 val
的元素。nums
的其余元素和 nums
的大小并不重要。
返回 k。
示例 1:
输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2,_,_]
解释:你的函数函数应该返回 k = 2, 并且 nums 中的前两个元素均为 2。
你在返回的 k 个元素之外留下了什么并不重要(因此它们并不计入评测)。
示例 2:
输入:nums = [0,1,2,2,3,0,4,2], val = 2
输出:5, nums = [0,1,4,0,3,_,_,_]
解释:你的函数应该返回 k = 5,并且 nums 中的前五个元素为 0,0,1,3,4。
注意这五个元素可以任意顺序返回。
你在返回的 k 个元素之外留下了什么并不重要(因此它们并不计入评测)。
第一种解法:暴力解法
即第一个for循环遍历数组一一寻找与之相同的数,第二个for循环使数组整体前移。
我第一次写的代码是有问题的,如下:
class Solution {
public int removeElement(int[] nums, int val)
{
int num = nums.length;
int k = num;
for(int i = 0; i < num ;i++)
{
if(nums[i]==val)
{
k--;
for(int j =i;j<num-1;j++)
{
nums[j]=nums[j+1];
}
}
}
return k;
}
}
这个代码运行可能会导致删得不完全,因为我的i不管什么时候都是向前移动的,但是遇到nums[]==val时,假设i原本移到了第三个元素上,第三个被删掉了,原本的第四个元素现在成为了第三个元素,i却变成了4,它并没有来得及对这个新的第三个元素加以判断,所以代码需要改进,改进之后代码如下:
class Solution {
public int removeElement(int[] nums, int val)
{
int num = nums.length;
int k = num;
for(int i = 0; i < k;)
{
if(nums[i] == val)
{
k--;
for(int j = i; j < k; j++)
{
nums[j] = nums[j + 1];
}
// 注意:这里不需要 i++,因为我们已经将后面的元素前移,
// 下一次迭代会检查当前位置的元素。
}
else
{
i++; // 只有当当前元素不是 val 时,才递增 i。
}
}
return k;
}
}
第二种解法:双指针法
通过一个快指针和慢指针在一个for循环下完成两个for循环的工作。
class Solution
{
public int removeElement(int[]nums ,int val)
{
int slowindex =0;
for(int fastindex = 0;fastindex<nums.length;fastindex++)
{
if(nums[fastindex]!=val)
{
nums[slowindex]=nums[fastindex];
slowindex++;
}
}
return slowindex;
}
}