【原创】90% 的人都不会做的一道笔试题

数组操作的题目，有的确实比较容易，但并非每个问题都是如此。今天就来看道 90%的人都不会做的笔试题。请看题：

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例：

给定 nums = [3,1,4,3], val = 3，函数应该返回新的长度 2，并且 nums 中的前两个元素均为 2。

你不需要考虑数组中超出新长度后面的元素。

请暂时不要看下面，先思考一下，有没有解答思路。

先

思

考

再

看

！

！

！

解题思路：

现在考虑数组包含很少的要删除的元素的情况。例如，num=[1,2,3,5,4]，Val=4。按照很多人的思维会对前四个元素进行复制操作，其实咱们没必要对前四个元素做复制操作。另一个例子是 num=[4,1,2,3,5]，Val=4。也似乎没有必要将 [1,2,3,5][1,2,3,5] 这几个元素左移一步，因为问题描述中提到元素的顺序可以更改。

实现

当我们遇到 nums[i] = val 时，我们可以将当前元素与最后一个元素进行交换，并释放最后一个元素。这实际上使数组的大小减少了 1。

请注意，被交换的最后一个元素可能是您想要移除的值。但是不要担心，在下一次迭代中，我们仍然会检查这个元素。

最多遍历 n 步。在这个方法中，赋值操作的次数等于要删除的元素的数量。因此，如果要移除的元素很少，效率会更高。

代码如下：

public static int removeElement(int[] nums, int val) { if(nums == null){ return 0; } int i = 0; int n = nums.length; while (i < n) { //如果相等 //（注意当当前元素是 5 的时候，此时 i 并没有加 1） if (nums[i] == val) { //当前这个位置 nums[i] = nums[n - 1]; //数组大小减 1 n--; } else { i++; } } return n; }

测试

场景 1：数组为 null；

public static void main(String[] args) { int[] nums =null; int val = 5; System.out.println(removeElement(nums, val)); }

结果

场景 2：数组中只有一个 5

public static void main(String[] args) { int[] nums ={5}; int val = 5; System.out.println(removeElement(nums, val)); }

结果

场景三：数组中有一个 5

public static void main(String[] args) { int[] nums ={1, 3, 5, 6}; int val = 5; System.out.println(removeElement(nums, val)); }

结果

场景 4：数组中有多个 5

public static void main(String[] args) { int[] nums ={1, 3, 5,5, 6}; int val = 5; System.out.println(removeElement(nums, val));}

结果：

场景 4：数组中没有 5

public static void main(String[] args) { int[] nums ={1, 3, 4, 6}; int val = 5; System.out.println(removeElement(nums, val));}

结果

场景 5：数组的最后一个元素是 5

public static void main(String[] args) { int[] nums ={1, 3, 4, 5}; int val = 5; System.out.println(removeElement(nums, val));}

结果

场景 6：数组中有多个 5，并且最后一个元素是 5

public static void main(String[] args) { int[] nums = {1, 3, 5, 6, 8, 9, 5, 5}; int val = 5; System.out.println(removeElement(nums, val));}

结果

最后一种场景：数组中全是 5

public static void main(String[] args) { int[] nums = {5, 5, 5, 5}; int val = 5; System.out.println(removeElement(nums, val));}

结果

ok，以上几种场景都涉及到了，并且结果正确。

还不明白的，可以看下图：