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

关注“Java后端技术全栈”**

回复“面试”获取全套大厂面试资料

数组操作的题目，有的确实比较容易，但并非每个问题都是如此。今天就来看道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 这几个元素左移一步，因为问题描述中提到元素的顺序可以更改。

实现

当我们遇到 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));
}

结果