力扣练习题-移除元素
移除元素
题目描述:
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。
假设 nums 中不等于 val 的元素数量为 k,要通过此题,您需要执行以下操作:
更改 nums 数组,使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
返回 k。
用户评测:
评测机将使用以下代码测试您的解决方案:
int[] nums = [...]; // 输入数组
int val = ...; // 要移除的值
int[] expectedNums = [...]; // 长度正确的预期答案。
// 它以不等于 val 的值排序。
int k = removeElement(nums, val); // 调用你的实现
assert k == expectedNums.length;
sort(nums, 0, k); // 排序 nums 的前 k 个元素
for (int i = 0; i < actualLength; i++) {
assert nums[i] == expectedNums[i];
}
如果所有的断言都通过,你的解决方案将会 通过。
示例 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 个元素之外留下了什么并不重要(因此它们并不计入评测)。
解题思路
1、初始化慢指针 slow:
slow 指针用于标记新数组的当前位置,初始值为 0,表示新数组从索引 0 开始填充。
2、快指针 fast 遍历数组:
fast 指针用于遍历整个数组,检查每个元素是否等于 val。
如果 nums[fast] != val,说明这个元素应该保留,于是将其复制到 nums[slow],并让 slow 指针向后移动一位。
如果 nums[fast] == val,则跳过该元素,fast 继续向后移动,slow 保持不变。
3、返回新数组的长度:
遍历结束后,slow 的值就是新数组的长度,因为 slow 记录了所有保留元素的个数。
代码示例
C++版
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int slow=0;
for(int fast=0;fast<nums.size();fast++)
{
if(val!=nums[fast])
{
nums[slow++]=nums[fast];
}
}
return slow;
}
};
Java版
class Solution {
public int removeElement(int[] nums, int val) {
int slow=0;
for(int fast=0;fast<nums.length;fast++)
{
if(val!=nums[fast])
{
nums[slow++]=nums[fast];
}
}
return slow;
}
}

浙公网安备 33010602011771号