用Python对颜色排序
假设我们有一个包含n个对象的数组。这些颜色是红色，白色或蓝色，将它们分类就位，以使相同颜色的对象相邻。因此，颜色按红色，白色和蓝色的顺序排列。在这里，我们将使用数字0、1和2分别代表红色，白色和蓝色。因此，如果数组类似于[2,0,2,1,1,0]，则输出将为[0,0,1,1,2,2]
为了解决这个问题，我们将遵循以下步骤-
设置低：=0，中：=0，高：=数组长度–1
而中期<=高
如果arr[mid]=0，则交换arr[mid]和arr[low]，并将low和mid增加1
否则，当arr[mid]=2时，交换arr[high]和arr[mid]，将high减1
否则增加1
示例(Python)
让我们看下面的实现以更好地理解-
class Solution(object):
def sortColors(self, nums):
low = 0
mid = 0
high = len(nums)-1
while mid<=high:
if nums[mid] == 0:
nums[low],nums[mid] = nums[mid],nums[low]
low+=1
mid += 1
elif nums[mid] == 2:
nums[high], nums[mid] = nums[mid], nums[high]
high-=1
else:
mid += 1
return nums
ob1 = Solution()print(ob1.sortColors([2,0,2,1,1,0]))输入值
[2,0,2,1,1,0]
输出结果
[0,0,1,1,2,2]