【算法日积月累】8-三路快排

【算法日积月累】8-三路快排

使用“双指针”实现的 partition 操作，把数组分成三个部分

分析清楚以后，我们就可以很轻松地写出代码：

def __partition_3(nums, left, right):
    p = nums[left]
    # lt 是 less than 的意思，表示严格小于
    lt = left
    # gt 是 great than 的意思，表示严格大于
    gt = right + 1
    # i 这个变量用于遍历数组中的标定点以后的元素
    i = left + 1
    # 注意循环可以继续的条件，为什么不可以取“=”
    while i < gt:
        if nums[i] < p:
            lt += 1
            nums[i], nums[lt] = nums[lt], nums[i]
            i += 1
        elif nums[i] == p:
            i += 1
        else:
            gt -= 1
            nums[i], nums[gt] = nums[gt], nums[i]
    # 想清楚，为什么交换 left 和 lt 
    nums[left], nums[lt] = nums[lt], nums[left]
    return lt, gt

def __quick_sort(nums, left, right):
    if left >= right:
        return
    lt, gt = __partition_3(nums, left, right)
    # 在有很多重复元素的排序任务中，lt 和 gt 可能会相距很远
    # 因此后序递归调用的区间变小
    # 递归的深度也大大降低了
    __quick_sort(nums, left, lt - 1) # 想清楚为什么这里右边界是 lt - 1
    __quick_sort(nums, gt, right)


def quick_sort(nums):
    __quick_sort(nums, 0, len(nums) - 1)

关于使用两个指针把数组分成 3 个部分，在 LeetCode 上有一个专门的问题，我们可以当成练习巩固一下。

练习

LeetCode 第 75 题：颜色分类

其实最容易想到的是计数：分别统计 0、1、2 出现的次数，然后再对数组重新赋值。但是要使用 3 个辅助空间，编写要遍历数组两次。我们完全可以使用这一节介绍的“双指针”的办法。

解法1：

class Solution:
    def sortColors(self, nums):
        """
        :type nums: List[int]
        :rtype: void Do not return anything, modify nums in-place instead.
        """
        counter = [0] * 3
        for num in nums:
            counter[num] += 1
        i = 0
        for idx, count in enumerate(counter):
            for _ in range(count):
                nums[i] = idx
                i += 1

解法2：

class Solution:
    def sortColors(self, nums):
        l = len(nums)

        # 循环不变量的定义：
        # [0,zero] 中的元素全部等于 0
        # [zero+1,i) 中的元素全部等于 1
        # [two,l-1] 中的元素全部等于 2
        zero = -1
        two = l
        i = 0  # 马上要看的位置

        while i < two:
            if nums[i] == 0:
                zero += 1
                nums[zero], nums[i] = nums[i], nums[zero]
                i += 1
            elif nums[i] == 1:
                i += 1
            else:
                two -= 1
                nums[two], nums[i] = nums[i], nums[two]