LeetCode75_颜色分类

1. 题目

给定一个包含红色、白色和蓝色，一共 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]

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

进阶：
你可以不使用代码库中的排序函数来解决这道题吗？
你能想出一个仅使用常数空间的一趟扫描算法吗？

2. 题解

from typing import List


class Solution:
    # 冒泡排序
    def sortColors(self, nums: List[int]) -> None:
        for i in range(len(nums) - 1):
            for j in range(i + 1, len(nums)):
                if nums[i] > nums[j]:
                    nums[i], nums[j] = nums[j], nums[i]

    # 其他排序
    def sortColors1(self, nums: List[int]) -> None:
        left, point, right = 0, 0, len(nums) - 1
        while point <= right:
            if nums[point] == 2:
                nums[point], nums[right] = nums[right], nums[point]
                right -= 1
                continue        # 替换后, point不允许加1
            if nums[point] == 0:
                nums[point], nums[left] = nums[left], nums[point]
                left += 1
            point += 1


if __name__ == "__main__":
    s = Solution()
    nums = [1,2,0]
    s.sortColors1(nums)