题目:
给你一个数组
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 个元素之外留下了什么并不重要(因此它们并不计入评测)。
提示:
0 <= nums.length <= 100
0 <= nums[i] <= 50
0 <= val <= 100
代码如下:
法一:双指针
class Solution {
public int removeElement(int[] nums, int val) {
int last = 0;
for(int first=0;first<nums.length;first++){
if(nums[first]!=val){
nums[last] = nums[first];
last++;
}
}
return last;
}
}
法二:双指针优化
class Solution {
public int removeElement(int[] nums, int val) {
int first = 0;
int last = nums.length-1;
while(first<=last){
if(nums[first]==val){
nums[first]=nums[last];
last--;
// System.out.println("first指针:"+first+" last指针:"+last);
}else{
first++;
}
}
return first;
}
}
解释:
在法一中,通过设置俩个指针都从左边开始依次向右移动,其中first作为快指针用于遍历整个数组,last作为慢指针用于最后生成没有给定元素的数组。当快指针遍历的元素不是给定元素时,直接将该元素赋值到慢指针数组里,反之则继续遍历。
在法二中,同样也是设置俩个指针,但是不同于法一的是俩个指针放置于首尾,当首指针向后移动的过程中,如果碰到指定元素,则将尾指针指的元素进行替换,一直到遍历完数组元素。
注意:
在法二中,注意last每次更新后的位置都是没有遍历的元素,因此需要while条件包括等于号。