当前位置: 代码迷 >> 综合 >> leetcode75. Sort Colors(颜色排序)
  详细解决方案

leetcode75. Sort Colors(颜色排序)

热度:57   发布时间:2023-11-17 00:51:32.0

题目要求

给定一个包含红色、白色和蓝色,一共 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 = -1i = 0while i < end:if nums[i] == 2: end -= 1nums[end], nums[i] = nums[i], nums[end]elif nums[i] == 0:start += 1nums[start], nums[i] = nums[i], nums[start]i = i + 1else:i = i + 1        

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

  相关解决方案