3.移除元素
题目描述: 给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于val的元素,并返回移除后数组的新长度。
示例:
给定 nums = [0,1,2,2,3,0,4,2],val = 2
函数应该返回新的长度 5,并且 nums 中的前五个元素为 0,1,3,0,4
注意这五个元素可为任意顺序,不需要考虑数组中超出新长度后面的元素
-
解法一:暴力法。将数据从前往后扫描,如果出现该位置的元素值与给定的值相等,就将下一位置的元素覆盖该位置元素,向前移一位。
public static int getNewLen1(int[] nums , int val){ int len = nums.length; int i = 0; if (nums == null) { return 0; } while (i < len - 1) { if (nums[i] == val) {//如果该位置的元素与val相等,则覆盖该位置元素 for (int j = i ; j < len - 1; j++) { nums[j] = nums[j + 1]; } len--; }else { i++; } } return len; }
-
解法二:双指针法。保留两个指针 i 和 j,其中 i 是慢指针,j 是快指针,都从0开始遍历。当 nums[j]与给定的值相等时,说明是我们要移除的元素,此时自增 j,再判断下一个数字是否等于该给定的值,如果还相等继续自增j,直到nums[j]不等于给定的值,此时将nums[j]赋值给nums[i],然后 i 和 j 同时自增。重复以上步骤,直到 j 到数组末尾,新数组长度为 i。
//双指针 public static int getNewLen2(int[] nums , int val){ int i = 0; for (int j = 0; j < nums.length; j++) { //如果nums[j]不等于给定数字val,则赋值、自增,继续向下遍历 if (nums[j] != val) { nums[i] = nums[j]; i++; } } return i; //返回的是i,不是i+1,这是因为j是从0开始的,不是从i+1开始的 }