颜色分类 python

# 给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，
# 并按照红色、白色、蓝色顺序排列。
# 
#  此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。 
# 
#  
#  
# 
#  
# 
#  示例 1： 
# 
#  
# 输入：nums = [2,0,2,1,1,0]
# 输出：[0,0,1,1,2,2]
#  

方法：

# 三指针：如果只有两种颜色，那么很容易想到一前一后两个指针向中间遍历，颜色不对就交换位置。三种颜色仍然可以这么做，
# 只不过要多一个指针，前后两个指针用来分隔已经排好的红色和蓝色，中间的指针来遍历元素：
# 如果是红色，那么和前指针交换，并两个一起向后移，前指针换过来的一定是白色的，因为中指针已经扫描过那些元素了
# 如果是白色，那么继续向后移
# 如果是蓝色，那么和后指针交换，后指针向前移，中指针不能后移，因为此时不确定换过来的元素是什么颜色

nums = [1,2,1,1,0,0,2,1,0]
l=m= 0
r = len(nums)-1
while m <= r:
    if nums[m] == 0:
        # 左指针一直指向1的第一个元素，中指针一直指向最后一个1的下一个元素
        nums[m], nums[l] = nums[l], nums[m]
        l += 1
        m += 1
    elif nums[m] == 1:
        m += 1
    else:
        nums[m], nums[r] = nums[r], nums[m]
        r -= 1
print(nums)