75. 颜色分类——双指针

1.题目

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

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

 

示例 1：

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

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

输入：nums = [0]
输出：[0]
示例 4：

输入：nums = [1]
输出：[1]
 

提示：

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

进阶：

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

2.自我思路及实现

统计每种颜色的数量重写数组

时间：N，一趟扫描
空间：0

    class Solution {
    public void sortColors(int[] nums) {
        int a = 0;
        int b = 0;
        int c = 0;
        for(int i = 0; i < nums.length; i++)
        {
            if(nums[i] == 0)
                a++;
            else if(nums[i] == 1)
                b++;
            else 
                c++;
        }
        for(int i = 0; i < a; i++)
        {
            nums[i] = 0;
        }
        for(int i = a; i < a + b; i++)
        {
            nums[i] = 1;
        }
        for(int i = a + b; i < nums.length; i++)
        {
            nums[i] = 2;
        }

    }
}

3.总结思路及实现

双指针

使用两个指针

• left代表0的结尾位置
• right代表2的开头位置

遍历数组

• 数字为0，与left位置数字交换位置，同时left向右移动一位
• 数字为1，不操作
• 数字为2，与right位置数字交换位置，同时right向左移动一位
  此时交换后的数字可能是0或2，如果不处理进行下一步，那么将会漏数字，因此，如果是0或2，那么回退一位，处理这个数字

时间：N
空间：1

class Solution {
    public void sortColors(int[] nums) {

        int left = 0;
        int 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--;
                }
            }
        }

    }
}

或一个指针交换0，一个指针交换1

4.相关知识

• 本题是经典的「荷兰国旗问题」，由计算机科学家 Edsger W. Dijkstra 首先提出。
• 如果颜色过多，那么直接排序是更好的选择