给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
思路:这是一道很简单的题,一个很简单的思路是设置一个变量x记录要删除的元素个数,遍历数组,如果遇到要删除的值则x加1,如果不是要删除的元素,则把该元素放在i-x的位置上(i为该元素本来的位置)。
代码:
public class RemoveElement {
public int removeElement(int[] nums, int val) {
int x = 0; //记录有多少个x
int len = nums.length;
int i = 0;
for(;i<len;i++) {
if(nums[i] == val)
x++; //记录val的数量
else
nums[i-x] = nums[i]; //向前移动
}
return i-x;
}
public static void main(String[] args) {
System.out.println(new RemoveElement()
.removeElement(new int[] {3,2,2,3}, 3));
System.out.println(new RemoveElement()
.removeElement(new int[] {0,1,2,2,3,0,4,2}, 2));
}
}
运行结果: