关注“Java后端技术全栈”**面试
回复“面试”获取全套大厂面试资料数据库
数组操做的题目,有的确实比较容易,但并不是每一个问题都是如此。今天就来看道90%的人都不会作的笔试题。请看题:后端
给你一个数组 nums 和一个值 val,你须要 原地 移除全部数值等于 val 的元素,并返回移除后数组的新长度。数组
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。缓存
元素的顺序能够改变。你不须要考虑数组中超出新长度后面的元素。函数
示例:测试
给定 nums = [3,1,4,3], val = 3,函数应该返回新的长度 2,而且 nums 中的前两个元素均为 2。你不须要考虑数组中超出新长度后面的元素。优化
请暂时不要看下面,先思考一下,有没有解答思路。spa
先线程
思
考
再
看
!
!
!
解题思路:
如今考虑数组包含不多的要删除的元素的状况。例如,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)); }
结果
ok,以上几种场景都涉及到了,而且结果正确。
推荐阅读