【python高效解题】Leetcode-75：颜色分类

我们的宗旨是：使用最简洁的代码，完成AC：）

题目介绍

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意：请不要使用代码库中的排序函数来解决这道题。

进阶：

你能想出一个仅使用常数空间的一趟扫描算法吗？

示例 1：

输入：nums = [2,0,2,1,1,0]
输出：[0,0,1,1,2,2]

示例 2：

输入：nums = [2,0,1]
输出：[0,1,2]

示例 3：

输入：nums = [0]
输出：[0]

示例 4：

输入：nums = [1]
输出：[1]

提示：

1. n == nums.length
2. 1 <= n <= 300
3. nums[i] 为 0、1 或 2

题目分析

该题是一个简单的排序问题，难点在于如何使用最少的指针，交换最少的次数。

最少应使用三个指针，分别指向每种颜色的尾巴，需要注意的是，“白色”、“红色”是从前往后排的，“蓝色”是从后往前排的：）

python代码

class Solution:
    def sortColors(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        red, white, blue = 0, 0, len(nums)-1
        while white <= blue:
            if nums[white] == 0:
                nums[red], nums[white] = nums[white], nums[red]
                red += 1
                white += 1
            elif nums[white] == 1:
                white += 1    
            else:
                nums[white], nums[blue] = nums[blue], nums[white]
                blue -= 1
        return nums