给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
示例:
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
进阶:
- 一个直观的解决方案是使用计数排序的两趟扫描算法。
- 首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。
- 你能想出一个仅使用常数空间的一趟扫描算法吗?
这个题提示的思路就是直接两趟扫描计数,这个比较简单,使用C++编写,另外一种可以利用只有三种颜色,红色开头,蓝色结尾,那么只需要设置两个指针,红指针指向头和蓝指针指向尾,如果碰到0就与红指针指向的数字交换,然后红指针加1,如果碰到2,那么与蓝色指针指向的数字交换,然后蓝色指针减1,需要注意的是,由于遍历都是从前往后,所以防止出现将0换到前面的情况,与蓝指针交换完之后需要退一步,否则就会出现跳过遍历的情况,另外,遍历时,直接遍历蓝色指针个数即可,这个数字是不停变换的,当当前步大于蓝色指针时,说明已经是排好序了。
C++源代码:
class Solution {
public:
void sortColors(vector<int>& nums)