【python-leetcode75-双指针】颜色分类

前端之家收集整理的这篇文章主要介绍了【python-leetcode75-双指针】颜色分类前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

问题描述:

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

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

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

示例:

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

进阶:

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

 

这种问题也被称为荷兰国旗问题,即将不同元素排列成一段段相同的。

解题:核心,有一个指针指向的值为1时,先让它保持不动。

@H_301_34@class Solution:
    def sortColors(self,nums):
        p0 = curr = 0
        p2 = len(nums) - 1

        while curr <= p2:
            if nums[curr] == 0:
                nums[p0],nums[curr] = nums[curr],nums[p0]
                print("curr==0-->",nums)
                p0 += 1
                curr += 1
            elif nums[curr] == 2:
                curr==2--> nums[p2],nums[curr]
                p2 -= 1
            else:
                curr += 1

                
print(Solution().sortColors([0,0]))

结果:

看一下图就明白了: 

 

猜你在找的Python相关文章