现在有若干个红、白、蓝三种颜色的球随机排列成一条直线。现在我们的任务是把这些球按照红、白、蓝排序,就类似于荷兰国旗一样。
这个问题我们可以将这个问题视为一个数组排序问题,这个数组分为前部,中部和后部三个部分,每一个元素(红白蓝分别对应0、1、2)必属于其中之一。
由于红、白、蓝三色小球数量并不一定相同,所以这个三个区域不一定是等分的,也就是说如果我们将整个区域放在[0,1]的区域里,由于三色小球之间数量的比不同(此处假设1:2:2),可能前部为[0,0.2),中部为[0.2,0.6),后部为[0.6,1]。我们的思路如下:将前部和后部各排在数组的前边和后边,中部自然就排好了。
具体方法如下:
设置两个标志位begin和end分别指向这个数组的开始和末尾,然后用一个标志位i从头开始进行遍历:
1)若遍历到的位置为0,则说明它一定属于前部,于是就和begin位置进行交换,然后i向前进,begin也向前进(表示前边的已经都排好了)。
2)若遍历到的位置为1,则说明它一定属于中部,根据总思路,中部的我们都不动,然后i向前进。
3)若遍历到的位置为2,则说明它一定属于后部,于是就和end位置进行交换,由于交换完毕后i指向的可能是属于前部的,若此时i前进则会导致该位置不能被交换到前部,所以此时i不前进。而同1),end向前进一。
具体代码如下:
class Solution {
public:
void sortColors(int A[], int n) {
int begin = 0;
int end = n-1;
int i = 0;
while(i<end+1)
{
if(A[i]==0)
{
swap(A[i], A[begin]);
begin++;
i++;
}
else if(A[i]==2)
{
swap(A[i], A[end]);
end--;
//i++;
}
else
i++;
}
}
};