1. 问题描述:
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
示例:
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
2. 思路分析:
① 其实题目还是不难理解的,主要的意思是将数组中的数字从小到大进行排序,一个很直接的想法是通过一层for循环对数组中的0,1,2三个数字出现的数字进行计数看一下各自出现的次数是多少,然后再使用一个for循环将对应次数的0,1,2数字填充到原数组中即可,这个方法的好处是简单明了容易解决问题,代码如下:
import java.util.Scanner;
public class Solution {
public void sortColors(int[] nums) {
int zero = 0, one = 0, two = 0;
for (int i = 0; i < nums.length; ++i){
if (nums[i] == 0){
zero++;
}else if (nums[i] == 1){
one++;
}else if (nums[i] == 2){
two++;
}
}
for (int i = 0; i < nums.length; ++i){
if (i < zero){
nums[i] = 0;
}else if (i - zero < one){
nums[i] = 1;
}else if (i - zero - one < two){
nums[i] = 2;
}
}
/* for (int i = 0; i < nums.length; ++i){
System.out.print(nums[i] + " ");
}*/
}
}
② 其实除了上面的解法之外,我们还可以对其进行优化,可以在原数组的基础上进行排序,使用一层for循环就可以将数组中的内部元素进行排序,可以借助于快速排序中多指针的移动的思想来实现排序,将1的元素往中间移动,2的元素往最右边移动,具体实现如下:
1)声明三个指针,一个左指针left,一个当前指向的正在扫描的指针current,另外一个是指向最右端的指针right,当current扫描到的是1的时候那么current直接往右移动,扫描到0的时候判断left指针指向的元素是否是1,假如是1那么交换left与current各自指向的元素,同时扫描到0的时候假如没有交换的时候需要将left指针往右边移动一位,这主要是将0的移到最左边,1的放在中间,当扫描到2的时候将current指针与right指针指向的元素交换,right往前移动一位
2)循环什么时候停止呢?条件为current的位置小于了right的位置,注意这里并不是!,否则有的情况会判断错误的吗,例如测试数据1,0,2就会发生错误,所以要注意一下,写代码的时候最好结合具体的例子写,看一下哪些是需要进行考虑的
3. 代码如下:
使用多指针进行移动代码:
import java.util.Scanner;
public class Solution {
public void sortColors(int[] nums) {
int left = 0, right = nums.length - 1, current = 0, temp;
while(current <= right){
if (nums[current] == 0){
/*current与left只存在着两种情况那么就是两者在同一位置要么就是current大于left位置*/
if (current > left && nums[left] == 1){
/*交换元素*/
temp = nums[current];
nums[current] = nums[left] ;
nums[left] = temp;
left++;
}
if (nums[left] == 0){
left++;
}
current++;
}else if (nums[current] == 1){
current++;
}else if (nums[current] == 2){
/*扫描到2*/
temp = nums[right];
nums[right] = nums[current];
nums[current] = temp;
right--;
}
}
}
}