Java实现-排颜色ii

来源:互联网 发布:淘宝网老是要重新登录 编辑:程序博客网 时间:2024/06/07 05:38

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

 注意事项

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

k <= n

样例

给出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;}}


原创粉丝点击