leetcode - 75. 颜色分类

难度 - 中等
原题链接 - 颜色分类

题目描述

给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。
我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
必须在不使用库内置的 sort 函数的情况下解决这个问题。

示例 1：
输入：nums = [2,0,2,1,1,0]
输出：[0,0,1,1,2,2]

示例 2：
输入：nums = [2,0,1]
输出：[0,1,2]

提示：
n == nums.length
1 <= n <= 300
nums[i] 为 0、1 或 2

在这里插入图片描述

双指针

该题为经典的荷兰国旗问题，由于题目本质是要我们将数分成三段。
具体步骤：
我们用L 和 R 卡住左边和边，左边放0，右边放1，中间位置放1，
遍历到0时放在左边，指针向右移动，继续比较
碰到1时，什么都不用做，继续向右比较，
碰到2时，放到右边。这样下来就分成了三层。
一次遍历就完成了。

代码演示

/** 
*分层
*/
public void sortColors(int[] nums) {
		//左边界，最开始左边没数据，用-1位置代替
      int l = -1;
      //右边没数据，初始化为数组长度
      int r = nums.length;
      //指针位置，比较
      int index = 0;
      while(index < r){
          if(nums[index] == 0){
              swap(nums,++l,index++);
          }else if(nums[index] == 1){
              index++;
          }else{
              swap(nums,index,--r);
          }
      }

    }
    //交换的方法
    void swap(int[] nums, int i, int j) {
      int temp = nums[i];
      nums[i] = nums[j];
      nums[j] = temp;
    }