75. Sort Colors刷题笔记

最新推荐文章于 2024-11-03 16:34:24 发布

waiting不是违停

最新推荐文章于 2024-11-03 16:34:24 发布

阅读量66

点赞数

分类专栏： LeetCode Python刷题文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/weixin_44737922/article/details/125768220

版权

LeetCode Python刷题专栏收录该内容

221 篇文章 1 订阅

订阅专栏

难点在于idx2是从最末端开始的

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