LeetCode第75题：颜色分类（中等）

LeetCode第75题：颜色分类（中等）

• 题目：给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。注意:不能使用代码库中的排序函数来解决这道题。
  
• 解题思路：原本这道题遍历一遍记下0，2的个数，再循环赋值依次就可以。但是题目要求仅使用常数空间的一趟扫描算法。我这样也算是扫描了一遍吧。。。

class Solution {
    public void sortColors(int[] nums) {
        int len = nums.length;
        int R = len-1;
        int L = 0;
        int p = nums[0];
        int k = 0;
        for(int i=0;i<len;i++){
            if(p==0 ){
                nums[L++] = 0;
                k++;
                if(k<=len-1) p = nums[k];
            }else if(p==2){
                p = nums[R];
                if(R>=0) nums[R--] = 2;
            }else if(p==1){
                if(k<len-1) p = nums[++k];
            }
        }
        
        for(int i=L;i<=R;i++){
            nums[i] = 1;
        }
    }
}

• 题解做法：官方给的做法跟我的差不多，只不过它是当前位置和左边或者是右边进行两两交换，而我的是将左边或右边的变成当前位置。

class Solution {
  /*
  荷兰三色旗问题解
  */
  public void sortColors(int[] nums) {
    // 对于所有 idx < i : nums[idx < i] = 0
    // j是当前考虑元素的下标
    int p0 = 0, curr = 0;
    // 对于所有 idx > k : nums[idx > k] = 2
    int p2 = nums.length - 1;

    int tmp;
    while (curr <= p2) {
      if (nums[curr] == 0) {
        // 交换第 p0个和第curr个元素
        // i++，j++
        tmp = nums[p0];
        nums[p0++] = nums[curr];
        nums[curr++] = tmp;
      }
      else if (nums[curr] == 2) {
        // 交换第k个和第curr个元素
        // p2--
        tmp = nums[curr];
        nums[curr] = nums[p2];
        nums[p2--] = tmp;
      }
      else curr++;
    }
  }
}

作者：LeetCode
链接：https://leetcode-cn.com/problems/sort-colors/solution/yan-se-fen-lei-by-leetcode/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。