给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。
示例2:给定 nums = [0,1,2,2,3,0,4,2], val = 2,
函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
注意这五个元素可为任意顺序。
方法一:
我的思路:
使用双指针,先定义一个变量len初始值为0,遍历数组,若数组中的元素和val的值相等,则跳过;若不相等,则把这个元素赋值给nums[len],len加1,;然后返回len则为移除之后数组的长度。
我的解法:
class Solution {
public int removeElement(int[] nums, int val) {
int len = 0;
for(int j = 0; j < nums.length; j++){
if(nums[j] != val){
nums[len] = nums[j];
len++;
}
}
return len;
}
}
执行用时 :1 ms, 在所有 Java 提交中击败了97.56%的用户
内存消耗 :35.6 MB, 在所有 Java 提交中击败了84.20%的用户
方法二:
我的思路:
**现在考虑数组包含很少的要删除的元素的情况。**例如,num=[1,2,3,5,4],Val=4。之前的算法会对前四个元素做不必要的复制操作。另一个例子是 num=[4,1,2,3,5],Val=4。似乎没有必要将 [1,2,3,5]这几个元素左移一步,因为问题描述中提到元素的顺序可以更改。
因此,当我们遇到 nums[i] = valnums[i]=val 时,我们可以将当前元素与最后一个元素进行交换,并释放最后一个元素。这实际上使数组的大小减少了 1。
我的解法:
class Solution {
public int removeElement(int[] nums, int val) {
int len = nums.length;
for(int i = 0; i < len;){
if(nums[i] == val){
nums[i] = nums[len-1];
len--;
} else {
i++;
}
}
return len;
}
}
执行用时 :1 ms, 在所有 Java 提交中击败了97.56%的用户
内存消耗 :36.1 MB, 在所有 Java 提交中击败了84.12%的用户