颜色排序

"""
颜色排序
给定一个包含红色、白色和蓝色，共x个元素的数组
原地对它们进行排序，使得相同颜色的元素相邻，
并按照红色、白色、蓝色顺序排列。
使用整数0、1、2分别表示红色、白色和蓝色
不使用内置的sort函数解决此问题。
"""

def colors_sort(numbers):
    """
    对数字列表进行排序，使得0出现在1之前，2出现在最后。
    该函数通过交换元素在列表中的位置来实现排序，无需额外的排序算法。

    参数:
    numbers: 一个包含0、1、2的整数列表。

    返回值:
    无返回值，函数直接在输入的列表上进行排序操作。
    """
    # 获取列表长度，用于后续的遍历
    length = len(numbers)
    # 初始化两个指针，point0指向最后一个0的位置，point1指向第一个1的位置
    point0 = 0
    point1 = 0

    # 遍历列表中的每个元素
    for i in range(length):
        # 如果当前元素是1，将其与point1指向的元素交换位置，然后point1后移
        if numbers[i] == 1:
            numbers[i], numbers[point1] = numbers[point1], numbers[i]
            point0 += 1
        # 如果当前元素是0，将其与point0指向的元素交换位置，然后point0和point1都后移
        elif numbers[i] == 0:
            numbers[i], numbers[point0] = numbers[point0], numbers[i]
            point1 += 1
            # 如果point0小于point1，说明有0在1的前面，需要将当前0与point1指向的1交换位置
            if point0 < point1:
                numbers[i], numbers[point0] = numbers[point1], numbers[i]
                point0 += 1
                point1 += 1

# 创建一个需要排序的列表
arr = [1, 2, 0, 1, 2, 0, 2]
# 调用排序函数
colors_sort(arr)
# 输出排序后的列表
print(arr)