现在有一个包含n个物体的数组,其中物体颜色为颜色为红色、白色或蓝色,请对这个数组进行排序,让相同颜色的物体相邻,颜色的顺序为红色,白色,蓝色。
我们用0,1,2分别代表颜色红,白,蓝
注意:
本题要求你不能使用排序库函数
扩展:
一个非常直接的解法是两步的计数排序的算法
首先:遍历一遍数组,记录0,1,2的数量,然后重写这个数组,先将0写入,再将1写入,再将2写入
你能给出一个只用一步,并且能在常数级空间复杂度解决这个问题的算法吗?
解题思路:(分别将0和2进行排序,等于0的排在最左边,等于2的排在最右边)三指针滑动,left指针从左边开始,记录最后一个为0的位置,right指针从最右边开始,记录最后一个数字2的位置,curl指针寻找到0的数字与最左边left指针交换,寻找到数字2的位置与最右侧指针交换,寻找到1的位置自增
public class Solution{
public static void sort(int[] array){
int left=0;
int right=array.length-1;
int curl=0;
while (curl<right){
if(array[curl]==0){
swap(array,curl++,left++);
}else if(array[curl]==2){
swap(array,curl,right--);
}else {
curl++;
}
}
}
public static void swap(int[] array, int curl, int i) {
int temp=array[curl];
array[curl]=array[i];
array[i]=temp;
}
}