Java实现-排颜色ii

给定一个有n个对象（包括k种不同的颜色，并按照1到k进行编号）的数组，将对象进行分类使相同颜色的对象相邻，并按照1,2，...k的顺序进行排序。

 注意事项

You are not suppose to use the library's sort function for this problem.

k <= n

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给出colors=[3, 2, 2, 1, 4]，k=4, 你的代码应该在原地操作使得数组变成[1, 2, 2, 3, 4]

挑战 

一个相当直接的解决方案是使用计数排序扫描2遍的算法。这样你会花费O(k)的额外空间。你否能在不使用额外空间的情况下完成？

class Solution {
    /**
     * @param colors: A list of integer
     * @param k: An integer
     * @return: nothing
     */
    public void sortColors2(int[] colors, int k) {
        // write your code here
        if(colors.length==0||colors==null){
			return;
		}
		for(int i=0;i<colors.length;i++){
			if(colors[i]>k){
				return;
			}
		}
		sort(colors);

    }
    
    private static void sort(int nums[]){
		int start=0;
		int end=nums.length-1;
		quickSort(nums, start, end);
	}
	
	private static void quickSort(int nums[], int start, int end){
		if(start<end){
			int p=partition(nums, start, end);
			quickSort(nums, start, p-1);
			quickSort(nums, p+1, end);
		}
	}
	
	private static int partition(int []nums, int start, int end){
		int left=start;
		int right=end;
		int pivot=nums[left];
		while(left<right){
			while(nums[right]>=pivot&&left<right){
				right--;
			}
			if(left<right){
				nums[left]=nums[right];
				left++;
			}
			while(nums[left]<pivot&&left<right){
				left++;
			}
			if(left<right){
				nums[right]=nums[left];
				right--;
			}
		}
		nums[left]=pivot;
		return left;
	}

}