LeetCode75（Sort Color）

问题

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.

对一个数组，有红，白，蓝三种颜色，以红，白，蓝的顺序对数组进行排序，以使相同的颜色相邻。

在这，我们使用0，1，2分别代表红，白，蓝三种颜色。

解答

这明显是一个排序问题，但排序的时间复杂度一般较高。比较快的排序方法的时间复杂度为O(nlog2(n))。然而，我们已经知道这个数组中数最多只有0，1，2三个或者其中某几个，传统的排序方法在时间复杂度上就不太好。可以先通过统计0，1，2的个数，然后根据个数对新数组进行赋值。

void sortColors(int A[], int n){
        int zero=0;
        int one=0;
        for(int i=0;i!=n;i++){
                if(A[i]==0){
                        zero++;
                }
                else if(A[i]==1){
                        one++;
                }
        }
        for(int i=0;i!=n;i++){
                if(i<zero){
                        A[i]=0;
                }
                else if(i>=zero && i<(zero+one)){
                        A[i]=1;
                }
                else{
                        A[i]=2;
                }
        }
}