“”"27. 移除元素:
给定一个数组 nums 和一个值 val,原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
元素的顺序可以改变。
示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。
“”"
class Solution:
#计算val出现的次数,for循环将其删除
def removeElement1(self, nums, val) -> int:
if val not in nums:
return len(nums)
for i in range(nums.count(val)):
nums.remove(val)
return len(nums)
# pop
def removeElement2(self, nums, val) -> int:
if val not in nums:
return len(nums)
i = 0
while i<len(nums):
if nums[i] == val:
nums.pop(i)
else:
i += 1
return len(nums)
#逆序遍历+remove
def removeElement3(self, nums, val) -> int:
for i in range(len(nums)-1,-1,-1):
if nums[i] == val:
nums.remove(i)
return len(nums)
#两个指针,small用来表示不重复的元素的下标,big用来遍历所有的元素
def removeElement4(self, nums, val) -> int:
small = 0
big = 0
while big<len(nums):
if nums[big] != val:
nums[small] = nums[big]
small += 1
big += 1
return small
#双指针
# 当前指针位置等于目标值时,和指向末尾的指针交换,并向左移动末尾指针
# 释放最后一个元素,如果不等,则当前指针向右移动一步,以此类推
# 边界条件是 前后指针相遇
def removeElement5(self, nums, val) -> int:
start = 0
end = len(nums)-1
while start <= end:
if nums[start] == val:
nums[start] = nums[end]
end -= 1
else:
start += 1
return start
if name == “main”:
L = [1,1,1,2,3,3,3,4,5,5,5,6,6,6]
s = Solution()
print(s.removeElement1(L,6))
print(s.removeElement2(L,6))
print(s.removeElement3(L,6))
print(s.removeElement4(L,6))
print(s.removeElement5(L,6))