python--lintcode148. 颜色分类-CSDN博客

描述

给定一个包含红，白，蓝且长度为 n 的数组，将数组元素进行分类使相同颜色的元素相邻，并按照红、白、蓝的顺序进行排序。

我们可以使用整数 0，1 和 2 分别代表红，白，蓝。

样例

给你数组 [1, 0, 1, 2], 需要将该数组原地排序为 [0, 1, 1, 2]。

挑战

一个相当直接的解决方案是使用计数排序扫描2遍的算法。

首先，迭代数组计算 0,1,2 出现的次数，然后依次用 0,1,2 出现的次数去覆盖数组。

你否能想出一个仅使用常数级额外空间复杂度且只扫描遍历一遍数组的算法？

这一题的挑战里已经说得很清楚了，就是要只扫描一遍且空间复杂度为O（1）。

思路是三个指针，left指针的左边一定全为0，right指针的右边全为2，i指针来遍历数组。

当i指针碰到0的时候，跟left指针交换，且left+1.

碰到1时，直接跳过下一个。

碰到2时，跟right指针交换，right-1。需要注意的是此时i不能加一，需要重新判断一次交换过来的数。

代码如下：

class Solution:
    """
    @param nums: A list of integer which is 0, 1 or 2
    @return: nothing
    """
    def sortColors(self, nums):
        # write your code here
        def swap(list,one,two):
            temp=list[one]
            list[one]=list[two]
            list[two]=temp
            return list

        left,right,i=0,len(nums)-1,0
        while(i<=right and left<right):
            if(nums[i]==1):
                i+=1
                continue
            if(nums[i]==0):
                swap(nums,left,i)
                i+=1
                left+=1
                continue
            if(nums[i]==2):
                swap(nums,right,i)
                right-=1
                continue
        return nums



s = Solution()
print(s.sortColors([0,2,2,2,2,1,0,1,0,0,0,1,0,2,0]))

这一题很简单，下一题才有意思。见我下一篇博客：https://blog.csdn.net/wenqiwenqi123/article/details/80819622