Leetcode[75] 包含0，1，2三种元素的数组排序

最新推荐文章于 2021-07-18 17:33:15 发布

傻傻的笑了

最新推荐文章于 2021-07-18 17:33:15 发布

阅读量2.6k

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/u010632630/article/details/79388768

leetcode 专栏收录该内容

1 篇文章

订阅专栏

本文探讨了LeetCode第75题，关于仅包含0、1、2三种元素的数组排序问题。文章指出，避免使用sort函数，可以采用更高效的算法。核心思路是对数组进行遍历，移动0到前面，2到后面，同时提供了Python实现代码，强调在处理2时需要特别注意的位置调整。

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

题目：0，1，2代表三种颜色，不采用sort函数进行排序。
分析：
传统的选择排序方法可以完成，复杂度为 $n^{2}$ 。本题目只有三种元素，因此应该有复杂度更低的算法。
核心思想：
进行一次遍历，如果是0，则往前移动，如果是2则往后。

python代码

class Solution(object):
    def sortColors(self,nums):
        start=0
        end=len(nums)-1
        ii=0
        while ii<=end:
            if nums[ii]==0:
                nums[ii]=nums[start]
                nums[start]=0
                start+=1
            if nums[ii]==2:
                nums[ii]=nums[end]
                nums[end]=2
                end-=1
                ii-=1
            ii+=1

在循环过程中，如果遇到元素等于2的情况，由于数字从end位置调换过来，因此需要再对该位置的数字进行判断，因此有ii-=1这一情况。也可以将对ii==0的判断放在后面。

class Solution(object):
    def sortColors(self, nums):
        start=0
        end=len(nums)-1
        i=0
        while i<=end:
            while (nums[i]==2 and i<=end):
                nums[i]=nums[end]
                nums[end]=2
                end-=1
            while (nums[i]==0 and i>=start):
                nums[i]=nums[start]
                nums[start]=0
                start+=1
            i+=1