关注我们获取更多计算机考研信息
? 今日习题
给你一个数组nums和一个值val,你需要原地移除所有数值等于val的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用O(1)额外空间并原地修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。
……⏰ 思考时间……
如果你想好答案了
请查看解题思路和代码实现
▼
·解题思路·
既然问题要求我们就地删除给定值的所有元素,我们就必须用 O(1)的额外空间来处理它。如何解决?我们可以保留两个指针i和j,其中i是慢指针,j是快指针。
当 nums[j] 与给定的值相等时,递增j以跳过该元素。只要 nums[j] != val ,我们就复制 nums[j]到 nums[i]并同时递增两个索引。重复这一过程,直到 j到达数组的末尾,该数组的新长度为 i。该解法与 删除排序数组中的重复项 的解法十分相似
·代码实现·
public int removeElement(int[] nums, int val) {int i = 0;for (int j = 0; j if (nums[j] != val) {
nums[i] = nums[j];
i++;
}
}return i;
}
·复杂度分析·
时间复杂度:O(n),假设数组总共有n个元素,i和j至少遍历2n 步。
空间复杂度:O(1)
微信关注“字节408考研”,
免费获取各院校计算机软件考研信息与专业课资料!