颜色分类
给定一个包含红,白,蓝且长度为 n 的数组,将数组元素进行分类使相同颜色的元素相邻,并按照红、白、蓝的顺序进行排序。
我们可以使用整数 0,1 和 2 分别代表红,白,蓝。
样例
给你数组 [1, 0, 1, 2]
, 需要将该数组原地排序为 [0, 1, 1, 2]
解题
暴力:两次快排 、扫描数组,记录0 1 2 的个数,更新数组
下面是一次快排 + 线性查找
对于数组 A,两个指针 low high
low指向 0的都跳过,结束的时候指向: 1或 2
high指向2的都跳过,结束的时候指向 :0 或 1
当是(1 ,0)时候,交换 low++ 使得0归位
当是(2,0)时候,交换high-- 使得 2归位
当是(2,1)时候,交换high-- 是的2归位
当时(1,1)时候,就比较复杂了
先左找,找到第一个0 和左1互换 low++
找不到0的时候从右1开始右找,找到第一个2的时候,和右1互换 high--
当左找不到2的时候,说明low high之间全是 1 结束
时间复杂度O(N)
class Solution { /** * @param nums: A list of integer which is 0, 1 or 2 * @return: nothing */ public void sortColors(int[] A) { // write your code here if(A == null) return; int len = A.length; int low = 0; int high = len - 1; while(low < high){ while(low <= high && A[low] == 0) low++; while(low <= high && A[high] ==2) high--; if(low < high ){ if(A[low] == 1 && A[high] ==0){ A[low] = 0; A[high] = 1; low++; }else if(A[low]==2 && A[high] ==0){ A[low] =0; A[high] =2; low++; high--; }else if(A[low]==2 && A[high] ==1){ A[low] =1; A[high] =2; high--; }else if(A[low] == A[high] && A[low] == 1){ // 左开始找0 int tmp0 = low + 1; while(tmp0 <= high && A[tmp0] != 0) tmp0++; // 找到了 0 和左1交换 if(tmp0 <= high && A[tmp0]==0){ A[low] = 0; A[tmp0] = 1; low++; }else { //没找到 0 // 右开始找2 int tmp2 = high - 1; while(tmp2 >= low && A[tmp2] != 2) tmp2--; // 找到 2 和右 1 交换 if(tmp2 >=low && A[tmp2] == 2){ A[high] = 2; A[tmp2] = 1; high--; }else{//没找到 0 也没找到 2 说明全是 1 结束 return; } } } } } } }