75. 颜色分类 Python

---

一、题目描述

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

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

必须在不使用库内置的 sort 函数的情况下解决这个问题。

示例 1

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

示例 2

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

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

二、代码

代码如下：

class Solution:
    def sortColors(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        def mao_pao(nums):
            num_len = len(nums)
            for j in range(num_len):
                sign = False
                for i in range(num_len - 1 - j):
                    if nums[i] > nums[i+1]:
                        nums[i], nums[i+1] = nums[i+1], nums[i]
                        sign = True
                if not sign:
                    break
        mao_pao(nums)
        print(nums)

---

三、解题思路

本题本质上是一个排序问题，且要求是原地算法，关于排序的算法有很多种，本题解采用最容易理解和实现的冒泡排序来解决本题。
由于本题的特殊性，将红色、白色和蓝色用0,1,2去表示，并且对其进行升序排序，冒泡排序刚好可以原地修改颜色列表，每一轮“冒泡”都会将当前待排序中最大的元素放在列表尾（升序）。