leetcode 75. Sort Colors 颜色分类 三色旗/荷兰国旗问题 快排思路

leetcode 75. Sort Colors 颜色分类 三色旗/荷兰国旗问题 快排思路


刷力扣不是为了AC而AC,而是嚼碎每一道题,面试时有思路可以表达。当然有些工程师更高尚的想法是为了算法思维、为了计算机算法领域的研究事业,鼓掌!

题目:

给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

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

示例:
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
进阶:一个直观的解决方案是使用计数排序的两趟扫描算法。
首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。你能想出一个仅使用常数空间的一趟扫描算法吗?

思路: 三色旗/荷兰国旗问题、快排思路: 保证left左面都是0,right右面都是2。k从左遍历到right即可,因为right右面是已经排好的了。如果nums[k]是1,就是中部的,不需要动,k+1继续排下一个;如果nums[k]是0,排在左边,left+1证明左边的都排完了,k+1接着排下一个;如果nums[k]是2,排在右边,right-1证明右边的都排完了,但是不确定调到这个位置的数的大小,所以要再排该位置的,k不变。扫描一遍即可。

细节:

  1. 坑:k的遍历小于right即可
  2. 坑:left需要+1,right不需要+1
  3. python列表元素交换位置:list[i],list[j] = list[j],list[i]

代码:

class Solution(object):
    def sortColors(self, nums):
        """
        :type nums: List[int]
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        left=0
        right=len(nums)-1
        k=0
        #保证left左面都是0,right右面都是2
        while k<=right: #right右边的是已经排完的了
            if nums[k]==1: #如果是1,就是中部的,不需要动,k+1继续排下一个
                k+=1
            elif nums[k]==0: #如果是0,排在左边,left+1证明左边的都排完了,k+1接着排下一个
                nums[left],nums[k]=nums[k],nums[left]
                left+=1
                k+=1
            else: #如果是2,排在右边,right-1证明右边的都排完了,但是不确定调到这个位置的数的大小,所以要再排该位置的
                nums[right],nums[k]=nums[k],nums[right]
                right-=1

本博客为原创作品,欢迎指导,转载请说明出处,附上本文链接,谢谢!

发布了20 篇原创文章 · 获赞 1 · 访问量 192

猜你喜欢

转载自blog.csdn.net/weixin_43973433/article/details/104992547