三色排序问题

问题描述

输入一个整型数组，每个元素在0~2之间，其中0，1，2分别代表红、白、蓝。现要求对数组进行排序，相同颜色的在一起，而且按红白蓝顺序先后排列。要求时间复杂度为O(n)。

 

问题分析

最容易想到的是排序，比如快排，归并，堆排等，但它们的时间复杂度为O(nlogn)，与题意不符。

第二种想到的是计数排序，扫描一遍过去，分别记录0，1，2的个数，然后再对数组进行赋值。时间复杂度为O(2n)，即O(n)，满足条件。

还有一种方法，只需扫描一遍数组即可，其充分利用了元素只有3种的特性：在扫描数组的时候，使用首尾俩个指针，分别指示0、1与1、2边界。比如源数组为{2， 2， 0， 0， 1， 1  }。

 

第一步：首指针p0，尾指针p1，i标识当前扫描位置，当前位置值为2，需要将其交换至尾指针p1位置，p1要向前移动一位，p0、i位置不变。

第二步：i位置值不为0、2，i要向后移动一位，p0、p1位置不变。