三路划分

问题：对一个只可能含有1、2、3三种值的数组，按升序排序。

另一种常见问法：有一个只可能含有红、绿、蓝三种颜色的序列，进行分类，使红色都位于最左边，蓝色都位于最右边。

解法I:使用计数排序，比较直观，不再说。

解法II：进行两次快排中的二路划分：

想将数组划分为1和非1两个区间，再将非1段划分为2和3两区间 。

解法III：仿照快排中用到的二路划分，进行三路划分。

设原数组为A， 有n个元素。

使用三个指针left， middle， right， 那么

A[0]...A[left-1]的值都为1

A[left]...A[middle-1]的值都为2

A[right+1]...A[n-1]的值都为3.

A[middle]...A[right]的值不确定，还未划分。

核心步骤：

1）判断A[right]是否为3，若是：right左移，继续查找；若不是，A[right]与A[middle]交换，转到2）。

2）判断A[middle]的值，若为1：与A[left]交换, left右移，middle右移；若为2：middle右移。

3）重复1）~2）直到middle > right

时间复杂度O（n），空间复杂度O（1）。

代码：

#include <vector>
#include <iostream>
using namespace std;

void SortTriSegment(vector<int> &v){
	int left = 0, middle = 0, right = (int)v.size() - 1;
	while(middle <= right){
		while(middle <= right && v[right] == 3)  --right;
		if(middle > right) break;
		swap(v[right], v[middle]);
		if(v[middle] == 1){
			swap(v[left], v[middle]);
			++left;
		}
		++middle;
	}
}

int main(){
	int arr[] = {3, 1, 2};
	vector<int> v(begin(arr), end(arr));
	SortTriSegment(v);
	for(auto p : v)
		cout << p <<" ";
	cout << endl;
	return 0;
}

问题扩展：

扩展1：如果是更多路划分呢？此算法是否仍适用？

扩展2：如果要求是稳定划分呢（即稳定排序）？