给定一个有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;
}
}