乐趣区

小李飞刀:python我来做题啦

好久不见
最近都在忙年底的年会,还沉迷于学习统计学,python 教程还停留在基础语法。但是仔细规划了下,还是要好好的瞄准目标前行。所以准备呢,实战走起来。
做题做题
无论如何,数据结构还是很重要的,所以准备每天刷 leetcode 的题目啦。慢慢的恢复数据结构知识,也实际的写一写 python。
第一题
27. 移除元素给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
class Solution:
def removeElement(self, nums, val):
“””
:type nums: List[int]
:type val: int
:rtype: int
“””
for i in nums:
if i == val:
nums.remove(i)
print(len(nums))
#这是第一版本的解法
但是测试出错了
末尾的 2 没有消除,数组中间的 2 是消除了的。为什么最后一个 2 会被错过呢。
不死心的又测试了下,果然是最后一个值会被错过。
class Solution:
def removeElement(self, nums, val):
“””
:type nums: List[int]
:type val: int
:rtype: int
“””
length = len(nums)
for i in range(length)
if nums[i] == val:
nums.remove(i)
print(len(nums))
#随手改了改
这个版本也是通过 for in 循环,根据下标进行查找,但是有个弊端,可能会越界。因为数组移除之后会变短,但是 length 没有更改。所以还是原来的根据 nums 直接 for in 更便捷,所以我们来找找为什么最后一个数值会被跳过吧~ 搜索了下发现有答案 python: 在 for 遍历 list 时使用 remove 出现的问题以及解析
因为 remove 删除元素后,指针数组所有数字前移,但是指针正常后移,所以,当到达最后一个数值的时候,指针早就不指向它了。既然知道了原因,我们来想想怎么解决这个问题吧!发现了个解决方案慎用 python 的 pop 和 remove 方法看来要慎用 remove 呢新方法来啦~
class Solution:
def removeElement(self, nums, val):
“””
:type nums: List[int]
:type val: int
:rtype: int
“””
for i in nums[:]:
if i == val:
nums.remove(i)
print(len(nums))

这是因为 x[:] 与 x 不是同一个 list,相当与把 x 的内存拷贝到一块新的内存,当对 x 做 remove 操作的时候,新的内存 list 并没有受影响。
然后成功提交啦,这是一个新的开始~ 开心用 4 行解决了一道题~python 你真棒!

退出移动版