python单链表实现荷兰国旗问题_Leetcode-75. 颜色分类 Sort Colors （荷兰国旗问题） -超详细python...

题目

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

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

注意:

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

链接：https://leetcode.com/problems/sort-colors/

Given an array with n objects colored red, white or blue, sort them in-place so that objects of the same color are adjacent, with the colors in the order red, white and blue.

Here, we will use the integers 0, 1, and 2 to represent the color red, white, and blue respectively.

Note: You are not suppose to use the library’s sort function for this problem.

Example:

Input: [2,0,2,1,1,0]

Output: [0,0,1,1,2,2]

思路及代码

荷兰国旗问题

用i，j，k分别代表三种数字的指针；

将所有数字都置为k代表的数字；

如果当前数字为i或者j，则置为j代表的数字，j指针往后移一位；

如果当前数字为i，则置为i代表的数字，i指针往后移一位。

最后，i结束的位置刚好是i所代表数字的个数，j结束的位置是i和j所代表数字之和，即i和j之间就是j代表数字的个数，剩下的是k。

class Solution:

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

"""

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

"""

i = j = 0

for k in range(len(nums)):

color = nums[k]

nums[k] = 2

if color < 2:

nums[j] = 1

j += 1

if color == 0:

nums[i] = 0

i += 1

其他方法：counting sort

复杂度

T = O ( n ) T = O(n)T=O(n)

S = O ( n ) S = O(n)S=O(n)