题目
“”"
给你一个数组 nums和一个值 val,你需要 原地 移除所有数值等于val的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参作任何拷贝
int len = removeElement(nums, val);
// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。
for (int i = 0; i < len; i++) {
print(nums[i]);
}
示例 1:
输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2]
解释:函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如,函数返回的新长度为 2 ,
而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也会被视作正确答案。
示例 2:
输入:nums = [0,1,2,2,3,0,4,2], val = 2
输出:5, nums = [0,1,4,0,3]
解释:函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
提示:
0 <= nums.length <= 100
0 <= nums[i] <= 50
0 <= val <= 100
通过次数755,693提交次
“”"
思路1
看见题目,第一反应就是把列表的val给删除了,遇见一个删除一个。进而考虑使用list的自带方法:
list.count():统计某个元素在列表中出现的次数。
list.remove():该方法没有返回值但是会移除列表中的某个值的第一个匹配项。
Python代码
class Solution(object):
def removeElement(self, nums, val):
"""
:type nums: List[int]
:type val: int
:rtype: int
"""
numbers = nums.count(val)
for i in range(numbers):
nums.remove(val)
return len(nums)
代码是能够通过的。但是写完提交再一想就觉得非常的low。
思路2
咋一想,用两层循环应该能搞定;外层循环遍历,内层循环遇见了就覆盖。
仔细想:
外层循环遍历数组元素,当遇见了要删除的元素时,后面的元素整体往前移动一位,把这个元素给覆盖住,相当于删除了这个元素,数组的长度也减1。
注意后面的数组向前移动完之后,外层遍历数组还要从这个位置开始遍历寻找是否是要删除的元素,(因为有可能存在连续两个val的情况。)
注意使用while循环的写法。
Python代码
# 外层循环遍历数组元素,内层循环遇见了val就后面的前移一位
class Solution(object):
def removeElement(self, nums, val):
"""
:type nums: List[int]
:type val: int
:rtype: int
"""
i = 0
size = len(nums)
while i < size:
if nums[i] == val: # 遇见了要删除的元素,后面的整体往前移动一位进行覆盖。
for j in range(i+1,size):
nums[j-1] = nums[j]
i -= 1 # 移动覆盖了找到的一个元素,相当于把元素删除了,指针要往后退一个,(while循环下有个 i+=1;)因为可能会遇见2个val连续的情况
size -= 1 # 根据题目意思:不需要考虑数组中超出新长度后面的元素。此时 size - 1
i += 1
return size
提交是能够过的,且速度比第一种要快。但是双循环还是很low的,有咩有更快的解法。
思路3
考虑使用双指针。
首先如果不考虑题目的要求“不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组”,那么很容易可以想到:再开辟一个列表,遍历原数组的元素,把不是val的元素添加到新列表当中。
def removeElement(self, nums, val):
"""
:type nums: List[int]
:type val: int
:rtype: int
"""
new_nums = []
for v in nums:
if v != val:
new_nums.append(v)
return len(new_nums)
在原地修改的前提下:考虑使用双指针来实现。
fast指针用来遍历数组,它是一直在走的(一直+1),终止条件就是遍历完数组。
slow指针也在走,但是当fast指针遇到val一次就暂停一次,遇见一次停一次,所以slow的值最后就是长度值。
当fast指针在遍历时,不是val时,就把值赋给slow,当遇到val时,slow暂停,fast继续往前走,相当于这次的赋值取消了,跳过了这次赋值。
相当于用一个slow的指针来记录“开辟的新列表的最后一位,即用来追加元素的位置”
代码分析:
外层循环,fast指针一直走,用来遍历数组,终止条件就是遍历完数组;
循环内,fast没遇到val时,就赋值,同时slow往前移动;当遇到val时,需要跳过此时赋值。
Python代码
def removeElement(nums, val):
slow = fast = 0
while fast < len(nums):
if nums[fast] != val:
nums[slow] = nums[fast] # 相当于快指针把所有不是val的值都赋值给了慢指针,即是遇到了val就跳过赋值
slow += 1
fast += 1
# 当 fast 指针遇到要删除的元素时停止赋值
# slow 指针停止移动, fast 指针继续前进
return slow