题目描述
现在有一个包含n个物体的数组,其中物体颜色为颜色为红色、白色或蓝色,请对这个数组进行排序,让相同颜色的物体相邻,颜色的顺序为红色,白色,蓝色。
我们用0,1,2分别代表颜色红,白,蓝
注意:
本题要求你不能使用排序库函数
扩展:
一个非常直接的解法是两步的计数排序的算法
首先:遍历一遍数组,记录0,1,2的数量,然后重写这个数组,先将0写入,再将1写入,再将2写入
你能给出一个只用一步,并且能在常数级空间复杂度解决这个问题的算法吗?
class Solution {
public:
void sortColors(int A[], int n) {
int front = 0, tail = n-1;
int i = 0;
while(i <= tail){
// 把0放到前面
if(A[i] == 0){
swap(A[front], A[i]);
i++;
front++;
}
// 把2放到后面
else if(A[i] == 2){
swap(A[tail], A[i]);
tail--;
}
else{
i++;
}
}
}
};
问题原因
- 第一次错是直接使用快排的思想,大的放右边,小的放左边,但没有考虑到,pivot为0时,会出现 021这样的错误情况。
- 第二次错误是把大的移到右边,交换来的大小是无法确定的,所以此时 i 的位置是不能动的