LeetCode刷题之路：75. 颜色分类

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

#最直观的思路

利用双指针思想，进行替换数字

class Solution:
    def sortColors(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        # left, right = 0, len(nums) - 1
        # cur = 0
        # while cur <= right:
        #     while cur <= right and nums[cur] == 2:
        #         nums[right], nums[cur] = nums[cur], nums[right]
        #         right -= 1
        #     if nums[cur] == 0:
        #         nums[left], nums[cur] = nums[cur], nums[left]
        #         left += 1
        #     cur += 1
        #     # print(nums)
        # return nums

        left, right = 0, 0
        for i in range(len(nums)):
            if nums[i] == 0:
                nums[left], nums[i] = nums[i], nums[left]
                left += 1
                if right < left:
                    right = left
            while right < i and nums[i] == 1:
                nums[right], nums[i] = nums[i], nums[right]
                right += 1
        return nums