版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
"""
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
一个直观的解决方案是使用计数排序的两趟扫描算法。
首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗?
"""
def sortColors(nums) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
count = [0, 0, 0]
for i in nums:
count[i] += 1
print(i,count[i])
print(count)
nums.clear()
for i in range(3):
for j in range(count[i]):
nums.append(i)
def sortColors02(nums):
zero = -1
two = len(nums)
i = 0
while i < two:
if nums[i] ==1:
i += 1
elif nums[i] == 0:
zero += 1
nums[zero],nums[i] = nums[i],nums[zero]
i += 1
else:
two -= 1
nums[two],nums[i] = nums[i],nums[two]
nums = [1,2,2,0]
sortColors02(nums)
print(nums)