【算法笔记】数组—双指针删除元素问题｜python

双指针——快慢指针

一个fast快指针，一个slow慢指针。我们让 fast 一直往后移动，相当于从数组第一个数遍历到最后一个数。我们指定一个值，当fast指向非指定值时，也就是——当且仅当我们发现 nums[fast] != val 的时候，我们把这个数拷贝到slow指向的位置，默认 slow 是从 0 开始的，然后 fast += 1 指向下一个位置。

这样就保证了前 slow 个数就是我们要的结果，不重复的数。

---

给定一个数组对其进行排序，删除等操作。

一、移动相同元素

给一个数组，将数组中的所有0 移动到末尾，保持非零元素的相对顺序，不复制原地对数组进行操作。

    def movezeroes(self, nums)：
        slow = fast = 0
        while fast < len(nums):
            if nums[fast] != 0:
                if slow != fast:
                    nums[slow], nums[fast] = nums[fast], nums[slow]
                slow += 1
            fast += 1

简单优化

把数组中所有非零元素，按顺序给数组的前段元素位赋值，剩下的全部直接赋值 0。定义一个 nums[ 0…i ]表示为非 0 元素的数组，在遍历数列时不断维护这个定义。

    def movezeroes(self, nums):
        n = len(nums)
        i = -1
        j = 0
        # nums[0....i]表示非0元素的数列,初始值i=-1
        while j <= n-1:
            if nums[j] != 0:
                i += 1
                nums[i] = nums[j]
            j += 1
        for z in range(i+1, n):
            nums[z] = 0

---

二、删除重复项

定义两个指针 fast 和 slow ，fast 快指针遍历数组的下标位置，慢指针指示下一个不同元素要填入的下标位置，初始两个指针都指向下标 1。

    def dele(self, nums: List[int]) -> int:
        if not nums:
            return 0
        n = len(nums)
        fast = slow = 1
        while fast < n:
            if nums[fast] != nums[fast - 1]:
                nums[slow] = nums[fast]
                slow += 1
            fast += 1
        return slow

在删除重复元素之后，每个元素只出现一次，并返回新的长度，遍历后nums[0] 到 nums[slow−1] 中的每个元素都不相同，新长度即为slow，返回slow 即可。

---

三、删除指定元素

给出一个数组 nums 和一个值 val，需要原地移除所有数值等于 val 的元素，并返回移除后数组的新长度。

class Solution:
    def removeElement(self, nums, val):
        if not nums:
            return 0
        slow = 0
        fast = 0
        while fast <= len(nums)-1:
            if nums[fast] != val:
                nums[slow] = nums[fast]
                slow += 1 #注意区分快慢指针，写反了会超时
            fast += 1
        return slow

跟第一个移动0一样，定义 nums[0…i] 为非 val 的数列，遍历整个数列
j代表快指针，i代表慢指针

    def removeElement(self, nums, val):
        n = len(nums)
        i = -1
        # 定义nums[0...i]为非val的数列
        j = 0
        while j <= n-1:
            if nums[j] != val:
                i += 1
                nums[i] = nums[j]
            j += 1
        return i+1