java算法，黄红蓝球依次排列

public class Test02 {
	public static void main(String[] args) {
		String[] balls = {"红","蓝","蓝","黄","红","黄","蓝","红","红","黄","红"};
		
		for(String string :sort(balls)) {
			System.out.println(string);
		}
		
	}
	//一堆球按照黄红蓝排序
	public static String[] sort(String[] balls) {
		//标记球的下标
		int k=0;
		//标记球的个数
		int num = 0;
		String tmp;
		for (int i = 0; i < balls.length; i++) {
			if("黄".contentEquals(balls[i])) {
				
				k=i;
				tmp = balls[num];
				balls[num] = "黄";
				balls[i] = tmp;
				num++;
			}
		}
		
		for(int j=num;j<balls.length;j++) {
			if("红".contentEquals(balls[j])) {
				
				k=j;
				tmp = balls[num];
				balls[num] = "红";
				balls[j] = tmp;
				num++;
			}
		}
		return balls;
	}
}