题目描述:
给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(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]
方法分析:
该题与26题的解题思路很类似,参考我的文章:https://blog.csdn.net/qq_30216191/article/details/81348501
我们需要考虑到:
- 给定的数组 nums 是一个无序数组,我们需要移除掉数组中所有值等于 val 的元素;所有我们要遍历数组,移除掉相应值即可
- 需要原地修改数组,那么我们就不能创建一个副本,而是只能在原数组上进行操作
下面我们就用具体的代码来实现吧。
代码实现:
var removeElement = function(nums, val) {
//遍历数组
for (var i = 0; i < nums.length; i++) {
//找出值等于 val 的元素
if (nums[i] == val) {
//删除之并向前移位
nums.splice(i,1);
i--;
}
}
//返回修改后数组的长度
return nums.length;
};
代码解析:
我们首先定义一个 removeElement 函数,其接收一个数组 nums 和一个目标值 val 作为参数;在函数内部,我们遍历数组,找出所有值为 val 的元素并移除掉;注意移除了元素后,数组的值就被更新了,所以我们需要把 i 减一,以确定不会跳位;最后返回修改后数组的长度即可。
该算法的时间复杂度为:O(n)
该算法的空间复杂度为:O(1)
相关链接:https://leetcode-cn.com/problems/remove-element/description/