【leetcode】数组-移除元素

好程序不脱发

已于 2022-11-18 10:15:25 修改

阅读量109

点赞数

分类专栏： python leetcode 文章标签： leetcode python 算法

于 2022-03-08 16:13:19 首次发布

本文链接：https://blog.csdn.net/ningmengzhihe/article/details/123353672

版权

python 同时被 2 个专栏收录

79 篇文章 1 订阅

订阅专栏

leetcode

66 篇文章 0 订阅

订阅专栏

Solution 1: 交换复制，首位双指针

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        # i记录当前元素下标，j记录被换元素的下标
        i, j = 0, len(nums)-1
        # 从前往后遍历
        while i <= j:
            # 如果当前元素==val，互换nums[i]和nums[j]
            if nums[i] == val:
                tmp = nums[i]
                nums[i] = nums[j]
                nums[j] = tmp
                j -= 1
            # 如果当前元素!=val，i++
            else:
                i += 1
        # 最后，nusm[0:i]是不等于val的元素数组，返回i
        return i

Solution 2: 拷贝复制，双指针都从头开始

class Solution(object):
    def removeElement(self, nums, val):
        """
        :type nums: List[int]
        :type val: int
        :rtype: int
        """
        
        slow = 0
        fast = 0
        k = len(nums)
        while fast < k:
            print(slow, fast, nums)
            if nums[fast]==val:
                fast = fast + 1
            else:
                print('hello')
                nums[slow] = nums[fast]
                slow = slow + 1
                fast = fast + 1
        return slow