LeetCode75-颜色分类

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

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

注意:
不能使用代码库中的排序函数来解决这道题。

示例:

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

进阶：

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

一、思路

就是排序问题嘛，它不让用库，自己写一个快排就好了

C++代码：

class Solution {
public:
	void sortColors(vector<int>& nums) {
		q_sort(nums, 0, nums.size() - 1);
	}
	void q_sort(vector<int>& nums, int low, int high) {
		if (low >= high)
			return;
		int q = partition(nums, low, high);
		q_sort(nums, low, q - 1);
		q_sort(nums, q + 1, high);
	}
	int partition(vector<int>& nums, int low, int high) {
		int x = nums[high];
		int i = low - 1;
		//int j = high - 1;
		for (int j = low; j < high; j++) {
			if (nums[j] < x) {
				i++;
				exchange(nums, i, j);
			}
		}
		exchange(nums, i + 1, high);
		return i + 1;
	}
	void exchange(vector<int>& nums, int a, int b) {
		int c = nums[a];
		nums[a] = nums[b];
		nums[b] = c;
	}
};

执行效率：