leetcode75. Sort Colors(颜色排序)

博客介绍了如何使用三路快排算法解决LeetCode 75题——颜色排序问题。通过设置三个指针,分别用于记录0、1、2的位置,实现原地排序,将红色、白色、蓝色数组元素按顺序排列,同时避免使用内置排序函数。详细解题思路和Python代码示例提供。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目要求

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

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

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

在这里插入图片描述

解题思路

本道题,我们使用三路快排三路快排,三个指针(分别记录:当前元素i,0的位置start,2的位置end)协同完成排序任务,我们这里的目标是:把0放到数组的前列,把2放在数组的尾部,根据这个目标我们可以进行一次遍历,当遇到0的时候,我们根据start的位置进行交换,遇到2的时候根据end的位置进行交换,这里要注意交换2之后不能马上移动指针,因为不知道换的是什么,是不是需要进一步的进行交换。当排好顺序0/2,那么1自然也就排好了。

主要代码python

class Solution(object):
    def sortColors(self, nums):
        """
        :type nums: List[int]
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        end = len(nums)
        start = -1
        i = 0
        while i < end:
            if nums[i] == 2: 
                end -= 1
                nums[end], nums[i] = nums[i], nums[end]
            elif nums[i] == 0:
                start += 1
                nums[start], nums[i] = nums[i], nums[start]
                i = i + 1
            else:
                i = i + 1        

原题链接:https://leetcode-cn.com/problems/sort-colors

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值