【leetcode】75.颜色分类

緈福的街口2025-05-22 14:28

文章目录

- - - [1. 单指针](#1. 单指针)
    - [2. 非原地（数组）](#2. 非原地（数组）)

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

1. 单指针

python 复制代码

class Solution(object):
    def sortColors(self, nums):
        """
        :type nums: List[int]
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        a = 0
        for i in range(len(nums)):
            if nums[i] == 0:
                nums[a], nums[i] = nums[i], nums[a]
                a += 1
        
        for i in range(a, len(nums)):
            if nums[i] == 1:
                nums[a], nums[i] = nums[i], nums[a]
                a += 1
        return nums

2. 非原地（数组）

python 复制代码

class Solution(object):
    def sortColors(self, nums):
        """
        :type nums: List[int]
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        red = 0
        white =0
        blue = 0
        for color in nums:
            if color == 0:
                red += 1
            elif color == 1:
                white += 1
            else:
                blue += 1
        nums[:red] = [0] * red
        nums[red: red + white] = [1] * white
        nums[red + white: ] = [2] * blue