一、 题目
给一个数组包含n个物体,有蓝色、红色和白色三种颜色,把他们分类并按照红、白、蓝的顺序排列,我们用0、1、2来表示红白蓝的颜色
注解:很容易想到遍历两遍数组得到三个数的数目,再覆盖,但是请只遍历一遍数组来解决。
二、 分析
很简单,题目的意思其实就是让对一个数组排序,数组中的元素只有0、1、2,并且要求只能遍历一遍数组,常数空间复杂度。此时我们可以想到借助于快速排序的partition思想,以1为中间枢纽对数组进行划分,使0在数组的左边,2在数组的右边,1在数组的中间。
class Solution { public: void sortColors(int A[], int n) { int end = -1; int start = n; int i=0; //end是放0那部分的尾部索引,start是放2那部分的首部索引 //碰到0放到end+1处,碰到2放到start-1处,碰到1指针直接后移 while(i<start){ if(A[i] == 0 && i!=++end) swap(A[end],A[i]); else if(A[i] == 2 && i!= --start) swap(A[start],A[i]); else i++; } } };
时间: 2024-11-03 22:25:38