给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
示例:
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
进阶:
一个直观的解决方案是使用计数排序的两趟扫描算法。
首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗?
分析
最开始比较暴力的方法是遍历两次,第一次把零排序排好,第二遍把 1 放在对应的位置上,剩下的就是 2 了,我们可以创建一个初始位置,在没移动一次 0 ,初始位置就加一,在第二次遍历的时候从出事位置开始,见 1 则调换位置。
或者就如同题目中给定的方法,第一次遍历记录下每个元素的个数,然后直接重新写这个数组中的数字,但这样能解题,但无法满足进阶要求,于是,我们可以使用双指针的办法。
我们可以设定两个指针放在头尾的位置,那么头尾的位置就是 0 和 2 这两个数字,那就数组从开头遍历,如果出现 0 就和开头换位置,如果出现2,就和尾指针换位置。但这样可能出现一个问题,就是可能出现指针滑动出现偏差的问题。
即和尾部进行调换以后,指针继续向前而漏掉了元素,如图示两种案例: