LeetCode75. 颜色分类

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

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

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

示例:

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

思路1：

采用快速排序思路，建立一个3行num.length+3列的二维数组，遍历原数组依次将对应颜色放入二维数组中，然后将二维数组中的元素按照先行后列的顺序赋给原数组，即可达到目的。

代码1：

public class SortColors75quick {
	public static void sortColors(int[] nums) {

		int k = 0;
		int length = nums.length;
		int[][] bucket = new int[3][length];
		int[] order = new int[length+2];
		for (int num : nums) {
			bucket[num][order[num]] = num;
			order[num]++;
		}
		for (int i = 0; i < 3; i++) {
				for (int j = 0; j < order[i]; j++) {
					nums[k] = bucket[i][j];
					k++;
				}
		}
	}

	@SuppressWarnings("static-access")
	public static void main(String[] args) {
		// TODO 自动生成的方法存根
		int[] A = new int[] { 1,2,0,0,2,1 };
		SortColors75quick test = new SortColors75quick();
		test.sortColors(A);
		for (int num : A) {
			System.out.println(num);
		}
	}
}

思路2：

在原数组中遍历判断小于1的放左边，大于1的放右边，等于1的不变所在位置；

代码2：

public void sortColors(int[] nums) {

		int less = -1;
		int more = nums.length;
		int i = 0;

		while (i < more) {
			if (nums[i] < 1) {
				swap(nums, ++less, i++);
			} else if (nums[i] > 1) {
				swap(nums, --more, i);
			} else {
				i++;
			}
		}
	}

	private void swap(int[] nums, int i, int j) {
		int t = nums[i];
		nums[i] = nums[j];
		nums[j] = t;
	}