LeetCode 75. 颜色分类

75. 颜色分类

   

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

进阶：

• 你能想出一个仅使用常数空间的一趟扫描算法吗？

思路：

本题是经典的「荷兰国旗 问题」：现在有若干个红、白、蓝三种颜色的球随机排列成一条直线。现在我们的任务是把这些球按照红、白、蓝排序。 

参考 三指针法：LeetCode题解

0，1，2 排序。一次遍历，若是0，则移动到表头，若是2，则移动到表尾。不用考虑1。0和2处理完，1还会有错吗？

• for循环判断条件：本来是 i <= len(nums)，但right指针右侧已经都是2了，没必要继续寻找。因此以越过右指针为终止条件，减少查找次数
• 判断p所在索引的数字
  • 找到0就与 left 头指针交换
  • 找到2就与 right 尾指针交换
  • 找到1则跳过

时间复杂度：O(N)，遍历数组

空间复杂度：O(1)，原地修改

// 参考 三指针法：https://leetcode.cn/problems/sort-colors/solutions/1947033/by-zyh698-e7u1/?languageTags=golang
// 荷兰国旗问题：0，1，2 排序。一次遍历，若是0，则移动到表头，若是2，则移动到表尾。不用考虑1。0和2处理完，1还会有错吗？
// left左侧都是0，right右侧都是2，左右侧都确定好了，那么中间的就自然全是1了
func sortColors(nums []int)  {
    left, right, i := 0, len(nums)-1, 0

    // 本来是 i <= len(nums)，但right指针右侧已经都是2了，没必要继续寻找。因此以越过右指针为终止条件，减少查找次数
    for i <= right {
        // 找到0就与左指针l交换
        if nums[i] == 0 {
            nums[left], nums[i] = nums[i], nums[left]
            left++
            i++
            // nums[l]已经在i向右遍历的过程中早就验证过了，所以i要加加（注意：left<=i，left在i的右侧）

        // 找到1就跳过    
        } else if nums[i] == 1 {
            i++

        // 找到2就与右指针r交换    
        } else { // if nums[i] == 2
            nums[i], nums[right] = nums[right], nums[i]
            right--
            // 此时i不用加1，因为nums[right]交换过来的数还没有验证，有可能是0或2
        }
    }
}