写点什么

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

用户头像
田维常
关注
发布于: 2020 年 10 月 25 日

数组操作的题目,有的确实比较容易,但并非每个问题都是如此。今天就来看道 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,以上几种场景都涉及到了,并且结果正确。


还不明白的,可以看下图:



发布于: 2020 年 10 月 25 日阅读数: 40
用户头像

田维常

关注

关注公众号:Java后端技术全栈,领500G资料 2020.10.24 加入

关注公众号:Java后端技术全栈,领500G资料

评论

发布
暂无评论
【原创】90%的人都不会做的一道笔试题