颜色分类

版权声明:本文为博主原创文章,如需转载,必须在转载处声明原作者,已经加入博客原链接地址。 https://blog.csdn.net/u013777382/article/details/79959896
       算法题:颜色分类
  给定一个包含红,白,蓝且长度为 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;
	}
}










阅读更多

没有更多推荐了,返回首页