L75 Middle 颜色分类 三路快排

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sort-colors

题目

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意:
不能使用代码库中的排序函数来解决这道题。

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

分析过程

• 初始状态
  
• 当值为1时，i++即可，生成第二幅图
  
  
• 如果是2，则取two前面的一个元素，与2进行交换，i不进行++,继续判断当前的i符合那种情况
  
  
• 如果值为0的话，将zero后面的一个1与该0，进行替换，同时，i++，判断下个元素的值
  
  
  

代码

public void sortColors(int[] nums) {

     int left  = -1;//[0,left]为0
     int right = nums.length;//[right,length-1]为2

     //此处的i是属于1的边界，不能超过2的起始边界
     for(int i= 0;i<right;){

         if(nums[i]==0){
             swap(nums,++left,i++);
         }else if (nums[i]==1){
             i++;
         }else if(nums[i]==2){
             swap(nums,i,--right);
         }
     }
 }

public static void swap(int[] arr,int s ,int e){
   int tmp = arr[s];
    arr[s]=arr[e];
    arr[e]= tmp;
}