移除元素(三 | 8.31)
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
思路:定义两个指针,一个是i,用来遍历数组,一个是index用来填充数组。
public class RemoveElement {
public static int removeElement(int[] nums, int val) {
int index = 0; // 指向待填充的位置
for (int i = 0; i < nums.length; i++) {
if (nums[i] != val) {
nums[index] = nums[i];
index++;
}
}
return index; // 返回新数组的长度
}
public static void main(String[] args) {
int[] nums = {3, 2, 2, 3, 4, 5, 3};
int val = 3;
int newLength = removeElement(nums, val);
// 输出移除指定元素后的数组
for (int i = 0; i < newLength; i++) {
System.out.print(nums[i] + " ");
}
}
}