Leetcode75. 颜色分类(三色国旗排列)

222 阅读1分钟

要求:

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

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

示例:

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

思路:

选用三个指针,p0,cur,p1。p0代表0的最右边界,p2代表2的最左边界,cur也就主要用于判断的移动指针。只要cur<=p2就进入循环。

  • 初始化:p0=cur=0;p2=nums.length-1;

  • 判断条件:

      1. nums[cur] = 0,交换nums[cur]和nums[p0],cur++,p0++
      2. nums[cur]=2,交换nums[cur]和nums[p2],cur不变,p2--
      3. nums[cur]=1,cur++
    

代码:

class Solution {
    public void sortColors(int[] nums) {
        int cur = 0;
        int p0 = 0;
        int p2 = nums.length-1;
        while (cur <=p2) {
            if(nums[cur] == 2) {
                int temp = nums[cur];
                nums[cur] = nums[p2];
                nums[p2--] = temp;
            } else if(nums[cur]==1) {
                cur++;
            } else {
                int temp = nums[cur];
                nums[cur] = nums[p0];
                nums[p0] = temp;
                p0++;
                cur++;
            }
        }
    }
}