[leetcode]Sort Colors

最新推荐文章于 2025-09-03 19:44:06 发布

weixin_33829657

最新推荐文章于 2025-09-03 19:44:06 发布

阅读量50

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法 c/c++

原文链接：https://yq.aliyun.com/articles/368131

本文介绍了一种针对红、白、蓝三色（用整数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.

基本思想：

遍历一遍记录0,1,2的个数，然后利用计数排序写出代码。

代码：

void sortColors(int A[], int n) {  //C++
        int rNum =0,wNum = 0, bNum =0; 
        for(int i = 0; i < n; i++){ 
            if(A[i] == 0) 
                rNum++; 
            else if(A[i] == 1) 
                wNum++; 
            else bNum++; 
        } 
         
        int rBegin = 0;  
        int wBegin = rNum;  
        int bBegin = rNum+wNum; 
         
        //ith element i < rNum  
        for(int i = 0; i < n; i++){ 
            if(i < rNum) 
                A[i] =0; 
            else if(i < n-bNum) 
                A[i] = 1; 
            else A[i] =2; 
        } 
}

one-pass algorithm

void sortColors(int[] A) {


    int i=-1, j=-1, k=-1;

    for(int p = 0; p < A.length; p++)
    {
        if(A[p] == 0)
        {
            A[++k]=2;
            A[++j]=1;
            A[++i]=0;
        }
        else if (A[p] == 1)
        {
            A[++k]=2;
            A[++j]=1;

        }
        else if (A[p] == 2)
        {
            A[++k]=2;
        }
    }

}

本文转自mfrbuaa博客园博客，原文链接：http://www.cnblogs.com/mfrbuaa/p/5059223.html，如需转载请自行联系原作者