LeetCode讲解篇之75. 颜色分类

文章目录

• 题目描述
• 题解思路
• 题解代码

题目描述

题解思路

我们可以将最终结果看成连续的三个区间，第一个区间内全是0，第二个区间内全是1，第三个区间内全是2

，其中这三个区间的长度都可以为0

我们可以将不断扩张

我们记录0区间的右边界的下一个位置和2区间的左边界的下一个位置
在遍历过程中
遇到0，交换当前元素和0区间右边界的元素，然后0区间的右边界向后扩张，如果左边界的索引大于等于当前遍历的元素的索引，则当前元素索引后移
遇到1，则索引后移
遇到2，交换当前元素和2区间左边界的元素，然后2区间的左边界向前扩张

题解代码

class Solution:
    def sortColors(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        left, right = 0, len(nums) - 1
        i = 0
        while i <= right:
            if nums[i] == 0:
                nums[i], nums[left] = nums[left], nums[i]
                left += 1
                if left >= i:
                    i += 1
            elif nums[i] == 1:
                i += 1
            else:
                nums[i], nums[right] = nums[right], nums[i]
                right -= 1