颜色分类

题目描述

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

示例

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

进阶

一个直观的解决方案是使用计数排序的两趟扫描算法。
首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗？

思路

三指针，左右两个指针维护0和2，中间的指针找到中间的0和2与两边的指针的元素进行交换。

实现

void sortColors(vector<int>& nums) {

	if (nums.size() < 1)
	{
		return;
	}

	int i = 0;
	int j = nums.size() - 1;
	int m = i;
	while (m < j)
	{
		while (i < nums.size() && nums[i] == 0)
		{
			i++;
		}
		while (j >= 0 && nums[j] == 2)
		{
			j--;
		}
		m = i;
		while (m < nums.size() && nums[m] == 1)
		{
			m++;
		}

		if (m <= j)
		{
			if (nums[m] == 0)
			{
				swap(nums[i], nums[m]);
			} else {
				swap(nums[j], nums[m]);
			}
		}
	}

}