题目描述
给你一个数组 nums
和一个值 val
,你需要移除所有数值等于 val
的元素,并返回移除后数组的新长度。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2]
解释:函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如,函数返回的新长度为 2 ,而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也会被视作正确答案。
示例 2:
输入: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。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
代码实现
方法一
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class Test3 {
public static void main(String[] args) {
// 原始数组
int[] arr = {0, 1, 2, 2, 3, 0, 4, 2};
int val = 2; // 需要移除的元素值
int num = removeE(arr, val); // 移除指定元素后的数组长度
int[] newArr = Arrays.copyOf(arr, num); // 根据移除后的数组长度创建新数组
List<Integer> list = new ArrayList<>();
for (int value : newArr) {
list.add(value);
}
System.out.println(num + ", nums = " + list); // 输出移除指定元素后的数组长度和数组内容
}
// 移除指定元素的方法
private static int removeE(int[] arr, int val) {
int front = 0; // 前指针
int rear = arr.length; // 后指针
while (front < rear) {
if (arr[front] == val) { // 如果当前元素等于要移除的元素值
arr[front] = arr[rear - 1]; // 将当前元素替换为数组末尾的元素
rear--; // 数组长度减一
} else {
front++; // 继续检查下一个元素
}
}
return front; // 返回移除指定元素后的数组长度
}
}
思路
使用了双指针的思想,移除数组中指定元素的值。
首先,在 main
方法中,初始化一个整型数组 arr
和一个整数 val
,其中 arr
数组包含需要被删除的元素。然后调用 removeE
方法,将 arr
数组和 val
值传递给该方法,以获取更新后的数组长度。接下来,使用 Arrays.copyOf
方法创建一个新数组,该数组将原数组中的前 num
个元素复制到新数组中,其中 num
就是通过 removeE
方法计算出来的需要保留的元素的数量。
在 removeE
方法中,初始化两个指针 front
和 rear
分别指向数组的头和尾。然后我们对数组进行遍历,如果当前元素等于要移除的元素值,就将当前元素替换为数组末尾的元素,然后尾指针 rear 左移一位。如果当前元素不需要被移除,则继续检查下一个元素。当头指针 front 和尾指针 rear 重合的时候,指针遍历完数组中所有的元素。
最后,我们返回另外一个指针 front
。这个指针的位置就是最终数组的长度。在 main
方法中,我们使用 Arrays.copyOf
和 List.add
方法创建一个新数组和列表,以存储保留的元素,并输出更新后的数组长度和内容。
总体来说,这段代码使用了简单而有效的双指针算法,时间复杂度为 O(n),空间复杂度也很小,只需要一个额外的列表就可以了。
方法二
public class Main {
public static void main(String[] args) {
int[] nums1 = {3, 2, 2, 3};
int val1 = 3;
int len1 = removeElement(nums1, val1);
System.out.println(len1); // 输出为 2
for (int i = 0; i < len1; i++) {
System.out.print(nums1[i] + " "); // 输出为 2 2
}
System.out.println();
int[] nums2 = {0, 1, 2, 2, 3, 0, 4, 2};
int val2 = 2;
int len2 = removeElement(nums2, val2);
System.out.println(len2); // 输出为 5
for (int i = 0; i < len2; i++) {
System.out.print(nums2[i] + " "); // 输出为 0 1 3 0 4
}
System.out.println();
}
public static int removeElement(int[] nums, int val) {
int i = 0; // 慢指针
// 使用快指针遍历数组
for (int j = 0; j < nums.length; j++) {
// 如果当前元素不等于val,则将其移动到慢指针所在位置,并更新慢指针
if (nums[j] != val) {
nums[i] = nums[j];
i++;
}
}
return i;
}
}
思路
removeElement
方法用于移除数组中所有值为 val
的元素,并返回新数组的长度。我们使用两个指针,一个是慢指针 i
,一个是快指针 j
。快指针遍历数组,如果当前元素不等于 val
,它一定是输出数组的一个元素,我们就将快指针 j 指向的元素复制到慢指针 i 位置,然后将两指针同时右移;如果快指针 j 指向的元素等于 val,它不能在输出数组里,此时慢指针不动,快指针右移一位。最后返回慢指针的值作为新的数组长度。