为国捐躯赴战场,丹心可并日争光。
给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
解答:java
public class Solution {public int removeElement(int[] nums, int val) {int n = nums.length;int j = 0; // 慢指针for (int i = 0; i < n; i++) {if (nums[i] != val) {nums[j] = nums[i];j++;}}return j;}public static void main(String[] args) {Solution solution = new Solution();int[] nums1 = {3, 2, 2, 3};int val1 = 3;int len1 = solution.removeElement(nums1, val1);System.out.println(len1); // 输出 2for (int i = 0; i < len1; i++) {System.out.print(nums1[i] + " ");}System.out.println(); // 输出 2 2int[] nums2 = {0, 1, 2, 2, 3, 0, 4, 2};int val2 = 2;int len2 = solution.removeElement(nums2, val2);System.out.println(len2); // 输出 5for (int i = 0; i < len2; i++) {System.out.print(nums2[i] + " ");}System.out.println(); // 输出 0 1 3 0 4}
}