给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
示例:
输入: [2,0,2,1,1,0] 输出: [0,0,1,1,2,2]
Java 语言实现:
采用计数排序的思想:
class Solution {
public void sortColors(int[] nums) {
//计数排序的思想
int[] a ={0,0,0};//分别存储0,1,2 的个数
for(int i = 0;i<nums.length;i++){
a[nums[i]]++;
}
int index = 0;
for(int j =0;j<a.length;j++){//遍历a数组
for(int i = 0;i<a[j];i++){
nums[index++] = j;//对nums数组重新赋值
}
}
}
}
进阶:
- 一个直观的解决方案是使用计数排序的两趟扫描算法。
首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。 - 你能想出一个仅使用常数空间的一趟扫描算法吗?
采用三路排序思想:
对nums数组分成三部分,分别是0部分:nums[0..zero] 1:nums[zero+1...two-1] 2:nums[two....length]
class Solution {
public void sortColors(int[] nums) {
int zero = -1; //nums[0...zero]
int two = nums.length; //nums[two..length-1]
int temp = 0;
for(int i = 0;i<two;){
if(nums[i] ==1){
i++;
}else if(nums[i] == 2){
temp = nums[i];
nums[i] = nums[--two];
nums[two] = temp;
}else{ //nums[i] ==0
temp = nums[++zero];
nums[zero] = nums[i];
nums[i] = temp;
i++;
}
}//end of for
}
}