《LeetCode力扣练习》第75题 颜色分类 Java

《LeetCode力扣练习》第75题 颜色分类 Java

一、资源

• 题目：

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

  进阶：

  你可以不使用代码库中的排序函数来解决这道题吗？
  你能想出一个仅使用常数空间的一趟扫描算法吗？
  

• 上代码（经过线上OJ测试）

  /**
   * Created with IntelliJ IDEA.
   *
   * @author : DuZhenYang
   * @version : 2022.03.01  18:01:48
   * description :
   */
  public class LeetCode {
      public void sortColors(int[] nums) {
          int n = nums.length;
          int p0 = 0;
          int p1 = 0;
  
          for (int i = 0; i < n; i++) {
              if (nums[i] == 1) {
                  swap(nums, p1, i);
                  p1++;
              } else if (nums[i] == 0) {
                  swap(nums, p0, i);
                  if(p0<p1){
                      swap(nums,p1,i);
                  }
                  p0++;
                  p1++;
              }
          }
      }
  
      private void swap(int[] nums, int p1, int i) {
          int temp = nums[p1];
          nums[p1] = nums[i];
          nums[i] = temp;
      }
  }