给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
nums = [0,1,2,2,3,0,4,2], val = 2
5, nums = [0,1,3,0,4]
解释:函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
public class Main{
public static void main(String[] args) {
//案例
int[] nums = new int[]{0, 1, 2, 2, 3, 0, 4, 2};
int val = 2;
//打印删除后数组元素个数
System.out.println(removeElement(nums, val));
//打印数组中元素
System.out.println(Arrays.toString(nums));
}
/**
* 按值删除
*
* @param nums 数组
* @param val 删除的目标值
* @return 返回删除后的元素个数
*/
public static int removeElement(int[] nums, int val) {
int index = 0;
//定义一个变量记录需要删除的元素个数
int cnt = 0;
//遍历nums
for (int i = 0; i < nums.length; i++) {
if (nums[i] == val) {
//计数
cnt++;
} else {
//不相等 将元素依次覆盖到数组的前面来
nums[index++] = nums[i];
}
}
//数组长度减去删除个数
return nums.length - cnt;
}
}