LeetCode-Sort Colors

最新推荐文章于 2021-12-23 23:39:18 发布

bsbcarter

最新推荐文章于 2021-12-23 23:39:18 发布

阅读量329

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/bsbcarter/article/details/44774559

leetcode 专栏收录该内容

220 篇文章

订阅专栏

本文介绍了一种有效的三色排序算法，该算法可以在一次遍历中完成对红、白、蓝三色（用0、1、2表示）的数组排序。通过使用两个指针分别指向已排序的0和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.

brute force的方法就是一遍把三种颜色都数一遍，然后再一遍一个一个赋值。走一遍的方法就是两个指针，一个在前面指排好的0的末尾，另一个在后面指排好的2的最前面，只要是遇到0 2 就swap，1就不用管。注意要用while，就是可能换过来的那个数字也要继续swap。同时两个while的先后顺序是有关系的，要先处理好2的情况，再swap所有0的情况。因为有可能从后面swap 2的时候，换过来一个0，但是不可能swap 0的时候从前面换过来一个2.因为current指的位置之前只可能有0和1.

public class Solution {
    public void sortColors(int[] A) {
        if (A ==null || A.length == 0)
            return;
        int first = 0;
        int last = A.length -1;
        for ( int i = 0; i <= last; i ++){
                while ( A[i] == 2 && i < last){
                    int temp = A[i];
                    A[i] = A[last];
                    A[last] = temp;
                    last --;
                }
                while ( A[i] == 0 && i > first){
                    int temp = A[i];
                    A[i] = A[first];
                    A[first] = temp;
                    first ++;
                }
        }
    }
}

还有一个解法没仔细研究感觉不太好想。

void sortColors(int A[], int n) {
    int n0 = -1, n1 = -1, n2 = -1;
    for (int i = 0; i < n; ++i) {
        if (A[i] == 0) 
        {
            A[++n2] = 2; A[++n1] = 1; A[++n0] = 0;
        }
        else if (A[i] == 1) 
        {
            A[++n2] = 2; A[++n1] = 1;
        }
        else if (A[i] == 2) 
        {
            A[++n2] = 2;
        }
    }
}