题目大意:
移除元素
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
给定 nums = [3,2,2,3], val = 3, 函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。 你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 nums = [0,1,2,2,3,0,4,2], val = 2, 函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。 注意这五个元素可为任意顺序。 你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参作任何拷贝 int len = removeElement(nums, val); // 在函数里修改输入数组对于调用者是可见的。 // 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。 for (int i = 0; i < len; i++) { print(nums[i]); }
如果想查看本题目是哪家公司的面试题,请参考以下免费链接: https://leetcode.jp/problemdetail.php?id=27
解题思路分析:
本题要求使用原地算法,也就是说不能新建一个数组来辅助解题。由于题目说明了不需要考虑最后数组内元素的顺序,因此我们可以使用以下逻辑来解题:
- 循环数组中每一个数,如果当前数字不等于val,继续
- 如果当前数字等于val,我们尝试将数组最后一位lastIndex的元素替换到当前位置。同时lastIndex减一,另外当前index不变,继续循环当前位置。
- 当前index如果大于 lastIndex ,可以退出循环。
- 返回结果为 lastIndex 加一(新数组长度)。
实现代码:java版本
public int removeElement(int[] nums, int val) { int lastIndex=nums.length-1; for(int i=0;i<=lastIndex;i++){ if(nums[i]==val){ nums[i]=nums[lastIndex]; lastIndex--; i--; } } return lastIndex+1; }
本题解法执行时间为0ms。
Runtime: 0 ms, faster than 100.00% of Java online submissions for Remove Element.
Memory Usage: 38.4 MB, less than 5.05% of Java online submissions for Remove Element.
Kotlin版本:
fun removeElement(nums: IntArray, `val`: Int): Int { var lastIndex = nums.lastIndex var i = 0 while(i <= lastIndex){ if(nums[i]==`val`){ nums[i]=nums[lastIndex] lastIndex-- }else{ i++ } } return lastIndex+1 }
本题解法执行时间为192ms。
Runtime: 192 ms, faster than 9.35% of Kotlin online submissions for Remove Element.
Memory Usage: 35.3 MB, less than 100.00% of Kotlin online submissions for Remove Element.
本网站文章均为原创内容,并可随意转载,但请标明本文链接如有任何疑问可在文章底部留言。为了防止恶意评论,本博客现已开启留言审核功能。但是博主会在后台第一时间看到您的留言,并会在第一时间对您的留言进行回复!欢迎交流!
本文链接: http://leetcode.jp/leetcode-27-remove-element-解题思路分析/