leetcode刷题之 75.颜色分类

leetcode刷题之 75.颜色分类

• 题目：
  给定一个包含红色、白色和蓝色，一共n个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。此题中，我们使用整数 0、1和 2 分别表示红色、白色和蓝色。
• 注意：
  不能使用代码库中的排序函数来解决这道题。
• 示例 :
  输入: [2,0,2,1,1,0]
  输出: [0,0,1,1,2,2]
• 进阶 :
  • 一个直观的解决方案是使用计数排序的两趟扫描算法。首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
  • 你能想出一个仅使用常数空间的一趟扫描算法吗？
• 代码1：

class Solution:
    def sortColors(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        a = nums.count(0)
        b = nums.count(1)
        c = nums.count(2)
        nums[:] = [0]*a+[1]*b+[2]*c
# 执行用时 :44 ms, 在所有 Python3 提交中击败了96.54%的用户
# 内存消耗 :12.7 MB, 在所有 Python3 提交中击败了99.85%的用户

• 算法说明：
  遍历一遍统计0,1,2的个数，然后给列表nums分别赋值即可。
• 代码2：

class Solution:
    def sortColors(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        p0,dangqian = 0,0
        p2 = len(nums) - 1
        while(dangqian <= p2):
            if nums[dangqian] == 0:
                nums[p0],nums[dangqian] = nums[dangqian],nums[p0]
                p0 += 1
                dangqian += 1
            elif nums[dangqian] == 2:
                nums[dangqian],nums[p2] = nums[p2],nums[dangqian]
                p2 -= 1
            else:
                dangqian += 1
# 执行用时 :88 ms, 在所有 Python3 提交中击败了7.29%的用户
# 内存消耗 :12.8 MB, 在所有 Python3 提交中击败了99.85%的用户

• 算法说明：
  用指针法，设置三个指针p0，p2，dangqian，分别代表0的位置、1的位置、当前位置的指针；用while判断，如果当前位置为0，将当前位置与代表0的指针位置进行交换；然后将0的指针和当前位置的指针后移一位；如果当前位置为2，将当前位置与代表2的指针位置进行交换；然后将2的指针前移一位；如果遇到1的位置，不参与交换，将当前位置后移一位；直到循环结束。算法流程：（DQ：代表当前位置）