LeetCode刷题（41）--Sort Color

zombee0

于 2017-11-18 10:54:36 发布

阅读量154

点赞数

CC 4.0 BY-SA版权

分类专栏：编程

本文链接：https://blog.csdn.net/zombee0/article/details/78567317

编程专栏收录该内容

72 篇文章

订阅专栏

本文介绍了一种名为1Pass的排序方案，该方案专门用于处理只有0、1、2三种数值的数组。通过将0向前移、2向后移的方式，实现了数组的有效排序，且排序过程中不返回任何内容，直接修改原数组。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1 Pass方案：
把0往前放，把2往后放。

class Solution(object):
    def sortColors(self, nums):
        """
        :type nums: List[int]
        :rtype: void Do not return anything, modify nums in-place instead.
        """
        p0 = 0
        p2 = len(nums) - 1
        cur = 0
        while cur <= p2:
            if nums[cur] == 0:
                nums[cur] = nums[p0]
                nums[p0] = 0
                p0 += 1
            elif nums[cur] == 2:
                nums[cur] = nums[p2]
                nums[p2] = 2
                p2 -= 1
                cur -= 1
            cur += 1
        return