2019-03-29 颜色分类

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

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

注意:

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

示例:

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

打眼一看就是个排序的题,因为前阵子做个冒泡,这次写个快排。很久没写了,虽然快排的逻辑自己清楚,但是怎么用代码实现出来还是废了很大的功夫。

快速排序的原理是二分法,找一个基准数,先从右到左找比它小的,再从左到右找比它大的,然后这两个数交换位置,一直找到两边相遇。然后吧基准数置换到相遇位置。然后,从相遇的位置把列表分成两个列表,递归上面的操作,直到结束。

代码如下:

class Solution:

    def sortColors(self, nums: List[int]) -> None:

        """

        Do not return anything, modify nums in-place instead.

        """

        start = 0

        end = len(nums)-1

        self.QuickSort(nums, start, end)


    def QuickSort(self,myList, start, end):

        # 判断low是否小于high,如果为false,直接返回

        if start < end:

            i, j = start, end

            # 设置基准数

            base = myList[i]

            while i < j:

                # 如果列表后边的数,比基准数大或相等,则前移一位直到有比基准数小的数出现

                while (i < j) and (myList[j] >= base):

                    j = j - 1

                # 如找到,则把第j个元素赋值给第个元素i,此时表中i,j个元素相等

                myList[i] = myList[j]

                # 同样的方式比较前半区

                while (i < j) and (myList[i] <= base):

                    i = i + 1

                myList[j] = myList[i]

            # 做完第一轮比较之后,列表被分成了两个半区,并且i=j,需要将这个数设置回base

            myList[i] = base

            # 递归前后半区

            self.QuickSort(myList, start, i - 1)

            self.QuickSort(myList, j + 1, end)

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。