Leetcode力扣秋招刷题路-0075

从0开始的秋招刷题路，记录下所刷每道题的题解，帮助自己回顾总结

75. 颜色分类（Mid）

• 给定一个包含红色、白色和蓝色、共 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

• 双指针

  • 与移动零思路类似
  • 把0移动到最前面，把2移动到最后面，若移动2之后交换的数字不是1，指针回退

  class Solution {
      public void sortColors(int[] nums) {
          int left = 0, right = nums.length - 1;
          for (int i = 0; i <= right; ++i) {
              if (nums[i] == 0) {
                  nums[i] = nums[left];
                  nums[left] = 0;
                  left++;
              }
              else if (nums[i] == 2) {
                  nums[i] = nums[right];
                  nums[right] = 2;
                  right--;
                  if (nums[i] != 1) i--;
              }
          }
      }
  }