leetcode学习笔记（75. 颜色分类）

本文已参与「新人创作礼」活动，一起开启掘金创作之路。

本题考察数组相关知识，为中等题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

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

代码

class Solution:
    def sortColors(self, nums: List[int]) -> None:
        nums.sort()
        
        return nums

双指针

class Solution:
    def sortColors(self, nums: List[int]) -> None:
        n = len(nums)
        i = 0
        p0, p2 = 0, n-1 
               
        while i <= p2:
            while i <= p2 and nums[i] == 2:
                nums[i], nums[p2] = nums[p2], nums[i]#把2换到最后
                p2 -= 1
            if nums[i] == 0:
                nums[i], nums[p0] = nums[p0], nums[i]#把0换到最开头
                p0 += 1
            i += 1

        return nums

解析

出题人预判了我这种人的存在，哈哈哈哈哈哈。本质上就是一个不用函数的排序问题，而且比一般的排序还要简单，只有三个数字！所以不需要用复杂的冒泡，二分啥的
具体方法：
1.用一个while循环
2.借助双指针P1指向第一个位置，p2指向最后一个位置，i等于2的放最后p2减1，等于0的放最开头p1加1
3.结束while循环的时候这个数列也就排好了

来源：力扣（LeetCode） 链接：leetcode.cn/problems/so… 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。