[LeetCode] Sort Colors 颜色排序

最新推荐文章于 2025-08-13 21:50:19 发布

转载最新推荐文章于 2025-08-13 21:50:19 发布 · 292 阅读

文章标签：

#leetcode #sort colors

经典题目专栏收录该内容

47 篇文章

订阅专栏

本文介绍了一种特殊的排序问题——三色排序，即在一个包含红色、白色和蓝色对象的数组中，仅使用整数0、1和2来表示颜色，并将它们按红、白、蓝的顺序排列。文中给出了两种解决方案，一种是通过计数排序进行两遍遍历，另一种是在常数空间内仅用一次遍历来完成排序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given an array with n objects colored red, white or blue, sort them so that objects of the same color are adjacent, with the colors in the order red, white and blue.

Here, we will use the integers 0, 1, and 2 to represent the color red, white, and blue respectively.

Note:
You are not suppose to use the library's sort function for this problem.

click to show follow up.

Follow up:
A rather straight forward solution is a two-pass algorithm using counting sort.
First, iterate the array counting number of 0's, 1's, and 2's, then overwrite array with total number of 0's, then 1's and followed by 2's.

Could you come up with an one-pass algorithm using only constant space?

这道题的本质还是一道排序的题，题目中给出提示说可以用计数排序，需要遍历数组两遍，那么先来看这种方法，因为数组中只有三个不同的元素，所以实现起来很容易。

- 首先遍历一遍原数组，分别记录0,1,2的个数
- 然后更新原数组，按个数分别赋上0，1，2

C++ 版：

class Solution {
public:
    void sortColors(int A[], int n) {
        int count[3] = {0}, idx = 0;
        for (int i = 0; i < n; ++i) ++count[A[i]];
        for (int i = 0; i < 3; ++i) {
            for (int j = 0; j < count[i]; ++j) {
                A[idx++] = i;
            }
        }
    }
};

题目中还要让只遍历一次数组来求解，那么我需要用双指针来做，分别从原数组的首尾往中心移动。

- 定义red指针指向开头位置，blue指针指向末尾位置

- 从头开始遍历原数组，如果遇到0，则交换该值和red指针指向的值，并将red指针后移一位。若遇到2，则交换该值和blue指针指向的值，并将blue指针前移一位。若遇到1，则继续遍历。

Java 版：

class Solution {
    
    public void swap(int[] nums, int a, int b){
        int tmp = nums[a];
        nums[a] = nums[b];
        nums[b] = tmp;
    }
    
    public void sortColors(int[] nums) {
        
        int r = 0, b = nums.length - 1;
        for (int i = 0; i <= b; ++i){
            if (nums[i]==0){
                swap(nums,i,r++);
            }else if(nums[i]==2){
                swap(nums,i--,b--);
            }
        }
        
        
    }
}

Python3 版：

class Solution:
    def sortColors(self, nums):
        """
        :type nums: List[int]
        :rtype: void 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]
                white += 1
                red += 1
            elif nums[white] == 1:
                white += 1
            else:
                nums[white], nums[blue] = nums[blue], nums[white]
                blue -= 1