题目描述:
现在有一个包含n个物体的数组,其中物体颜色为颜色为红色、白色或蓝色,请对这个数组进行排序,让相同颜色的物体相邻,颜色的顺序为红色,白色,蓝色。
我们用 0, 1, 2分别代表颜色红,白,蓝
注意:
本题要求你不能使用排序库函数
扩展: 一个非常直接的解法是两步的计数排序的算法 首先:遍历一遍数组,记录0,1,2的数量,然后重写这个数组,先将0写入,再将1写入,再将2写入 你能给出一个只用一步,并且能在常数级空间复杂度解决这个问题的算法吗?
解题思路:
- 题目要求不能用库函数,也不能二次遍历,所以使用一次遍历的方法
- 设置两个变量 zeroindex 和 twoindex,分别指向开始和结尾,类似于希尔或者说是插入排序
代码如下:
public void sortColors(int A[], int n) {
int zeroindex = 0;
int twoindex = n - 1;
int i = 0;
// 一次遍历数组,在遍历的过程中完成迭代
while(i < twoindex + 1)
{
if(A[i] == 0)
{
swap(A[i],A[zeroindex]);
zeroindex++;
i++;
}
else if(A[i]==2)
{
// 从尾部换过来的数可能是 1,所以不对 i 进行迭代
swap(A[i],A[twoindex]);
twoindex--;
}
else
i++;
}
}