Leetcode做题日记：75. 颜色分类(PYTHON)

最新推荐文章于 2023-08-12 21:06:32 发布

i火柴天堂

最新推荐文章于 2023-08-12 21:06:32 发布

阅读量112

点赞数 1

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

本文链接：https://blog.csdn.net/weixin_44033136/article/details/88133894

版权

leetcode 专栏收录该内容

113 篇文章 0 订阅

订阅专栏

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意:

不能使用代码库中的排序函数来解决这道题。

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

进阶：

一个直观的解决方案是使用计数排序的两趟扫描算法。

首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗？

第一次的代码：

	l=len(nums)
        nums=nums+[1]*l
        i=0
        j=2*l-1
        k=l
        while i<l:
            if nums[i]==0:
                nums[k]=0
                k=k+1
            if nums[i]==2:
                nums[j]=2
                j=j-1
            i=i+1    
        nums=nums[l:]

返回的nums没变，我觉得没错呀，在nums后面修改列表，然后返回修改值
第二次的代码：
遍历一遍，变一下0和2位置，1不用管，0和2变了，1自然是正确的位置

	l=len(nums)
        i=0
        j=l-1
        k=0
        while k<=j:
            if nums[k]==0: #若是0，交换位置k,i
                nums[k],nums[i]=nums[i],nums[k]
                i=i+1 
                k=k+1 #k前面的数一定比k处的值小 
            elif nums[k]==2:
                nums[k],nums[j]=nums[j],nums[k]
                j=j-1 #不改变k是因为有可能从后面换一个2过来
            else:
                k=k+1

36ms，排名11%
最标准的解法因该是统计0，1，2的频率，再修改nums
第三次代码：

	count=[0]*3
        for i in nums:
            count[i]+=1
        k=0    
        for i in range(3):#输出0，1，2
            for j in range(count[i]): #count[i]是i的频率
                nums[k]=i #k=0,1,2,3,...
                k=k+1