颜色分类

       算法题：颜色分类
  给定一个包含红，白，蓝且长度为 n 的数组，写一个函数实现将数组元素进行分类使相同颜色的元素相邻，
  并按照红、白、蓝的顺序进行排序。我们可以使用整数 0，1 和 2 分别代表红，白，蓝。


注意事项：
1、不能使用代码库中的排序函数来解决这个问题。
2、排序需要在原数组中进行。

格式：

第一行输入一个由 0，1，2 组成的数组，最后输出分类排序后的数组。

样例输入

[ 1，0，1，2 ]

样例输出

[ 0，1，1，2 ]

java版本的代码实现：

package cn.cat.algorithm;

import java.io.IOException;
import java.util.Arrays;
import java.util.Random;


public class ColorClassification {
	/** 
	 * @Description: 
	 * @author gwj
	 * @Created 2017年11月10日 上午10:36:46 
	 * @param args
	 * @throws IOException
	 */
	public static void main(String[] args) throws IOException {
		int RED = 0, WHITE = 1, BLUE = 2;
		int[] colors = new int[10];
		Random random = new Random();
		for (int i = 0; i < colors.length; i++) {
			colors[i] = random.nextInt(3);
		}
		System.out.println("排序前：" + Arrays.toString(colors));
		
		//先排序红色,然后排序白色,剩余的都是蓝色，不需要再排序
		int nextIndex = sortFront(colors, 0, RED);
		sortFront(colors, nextIndex, WHITE);
		System.out.println("排序后：" + Arrays.toString(colors));
	}
	
	/** 前向排序，将指定的颜色向前排序在一起。
	 * @Description: 
	 * @author gwj
	 * @Created 2017年11月10日 上午10:47:39 
	 * @param colors 颜色数组
	 * @param startIndex 起始标识
	 * @param color 颜色值
	 * @return 返回可使用的数组下一个索引值。
	 */
	static int sortFront(int colors[], int startIndex, int color) {
		for (int i = startIndex; i < colors.length; i++) {
			if (colors[i] == color) {
				int temp = colors[startIndex];
				colors[startIndex] = colors[i];
				colors[i] = temp;
				startIndex++;
			}
		}
		return startIndex;
	}
}