跳到主要内容

移除元素

一、题目描述​

题目: 给你一个数组 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 个元素之外留下了什么并不重要(因此它们并不计入评测)。

二、题解​

思路:​

  1. 设置快慢两个指针,快指针用于遍历数组搜索值为val的元素,慢指针用于记录下次交换非val元素的位置。

  2. 若快指针遍历到元素的值不为val,则与慢指针所指向位置的元素交换,且慢指针向右移动一个,单位计数count加1 。

  3. 快指针遍历完数组后,返回count即可

代码:​

class Solution {
public int removeElement(int[] nums, int val) {
int slow = 0;
int count = 0;
for(int i = 0;i < nums.length;i++){
if(nums[i] != val){
int temp = nums[i];
nums[i] = nums[slow];
nums[slow] = temp;
slow++;
count++;
}
}
return count;
}
}