LeetCode 075 Sort Colors

三色排序算法

最新推荐文章于 2025-09-08 16:25:29 发布

原创最新推荐文章于 2025-09-08 16:25:29 发布 · 701 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #面试笔试

LeetCode 专栏收录该内容

138 篇文章

订阅专栏

本文介绍了一种不使用库排序函数的三色数组排序算法。通过一次遍历和三个指针来实现0、1、2的有效排序，使得相同颜色的对象相邻，并按红、白、蓝的顺序排列。

题目

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.

数组排列颜色，一共3个颜色。

思路

1 桶排序的转换应用，基础。

2 关键点是最后设置新的数组的边界的时候，要注意，最好多测试一下，就清楚了。

3 最近突然想到新的方法，只要一遍遍历：有2个坐标，分别代表0,2 目前应该放的位子。还有个坐标cur，当cur碰到新的数，3中可能：

3.1 如果是1 ，继续前进

3.2 如果是0 ，和0应该放的位子交换,0的位子加一，目前的位子加一。（交换过来的要么是0，要么是1，无需继续检查）

3.3 如果是2，和2应该放的位子交换，2的位子减一。

代码

public class Solution {
    public void sortColors(int[] A) {
        if(A.length==0){
            return ;
        }
        int[] ref = new int[3];
        int n = A.length;
        for(int i=0;i<n;i++){
            int temp = A[i];
            ref[temp]++;
        }
        
        for(int i=0;i<ref[0];i++){
            A[i]=0;
        }
        for(int i=ref[0];i<ref[0]+ref[1];i++){
            A[i]=1;
        }
        for(int i=ref[0]+ref[1];i<ref[0]+ref[1]+ref[2];i++){
            A[i]=2;
        }
        
    }
}