Leetcode 27: 移除元素(python)
注:本人文章中有时有来自于各大佬的经验总结,仅供自身复习及学习,并无商业用途,若有时忘记标注作者及出处还请提醒,敬请见谅。
我的思路
由于题目要求原地移除元素,所以我们不能使用额外的辅助空间。因此,我们可以很直观地想到将所有值等于val的元素依次移到数组尾部。
具体的执行思路为:用一个变量count记录当前找到的值等于val的元素的个数,用n-count记录元素被挪到数组尾部的位置,如果当前尾部元素的值也为val,则将count加一,再往前一个位置进行寻找,直到找到尾部中下标大于i (这是为了避免将val又换回到数组头部)且值不等于val的位置为止,然后将两个元素进行交换。
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
count = 0
n = len(nums)
for i in range(n):
if (nums[i] == val) and (i < (n - count)):
count += 1
# 寻找到当前元素被交换到尾部的位置
# 如果当前尾部元素的值也为val,则向前一个位置进行寻找
while (nums[n - count] == val) and (n - count) > i:
count += 1
# 如果i位置后面的元素全部都为val
if (n - count)== i:break
# 交换元素
else:
temp = nums[n-count]
nums[n - count] = nums[i]
nums[i] = temp
return (n - count)
大佬思路:双指针
看题解的时候发现了双指针的妙用:
用一个指针a遍历数组,用另一个指针记录数组中最靠前的值为val的元素的位置。当a遍历到值不为val的元素时,将a和b对应位置上的元素交换位置,之后b自增1 。
图源代码随想录:
具体代码为:
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
a = 0
b = 0
while a < len(nums):
if nums[a] != val:
nums[b] = nums[a]
b += 1
a += 1
return b
作者:flying_du
链接:https://leetcode-cn.com/problems/remove-element/solution/python-shuang-zhi-zhen-da-fa-hao-a-quan-guo-zui-ca/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。