最简单的暴力解法,先统计各个颜色的次数,再遍历一次赋值,复杂度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;
}
}
}
但是很奇怪的是,第一种方法耗时居然比第二种要短,可能数据量少,优势没体现出来