Sort Colors

最简单的暴力解法,先统计各个颜色的次数,再遍历一次赋值,复杂度O(N)

public void sortColors(int[] A) {
        int red = 0,white=0,blue=0;
        int length = A.length;
        for(int p = 0; p < length; p++){
            if(A[p] == 0){
                red++;
            }else if (A[p] == 1){
                white++;
            }else if (A[p] == 2){
                blue++;
            }
        }
        for(int i = 0;i<length;i++){
            if(i<red){
                A[i]=0;
            }else if(i<(red+white)){
                A[i]=1;
            }else{
                A[i]=2;
            }
        }
    }
有没有只遍历一次的呢?下面这段代码就是,我没看得懂,暂且放着把

public 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;
            }
        }
    }
但是很奇怪的是,第一种方法耗时居然比第二种要短,可能数据量少,优势没体现出来


你可能感兴趣的:(Sort Colors)