给一个包含红色、白色和蓝色，共n个元素的数组nums，对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。用整数0、1、2分别表示红色、白色和蓝色。不要使用库的sort函数。

你可以使用一种叫做“计数排序”的非比较排序算法，配合一些辅助数组来实现这个目标。这里是一个基本的步骤描述：

1. 初始化三个长度为n的辅助数组count[0], count[1], count[2]，用于存储每个颜色元素的数量。
2. 遍历输入数组nums，对于每个元素，增加对应颜色数组的计数。
3. 更新一个新数组result，初始化为全零。从左到右遍历，每次取最小的颜色（优先红色），添加相应数量到result，同时递减count数组对应的计数。
4. 再次遍历count数组，将剩余未处理的颜色元素（先白后蓝）依次填充到result中。

这里是伪代码形式：

def sort_colors(nums):
    counts = [0, 0, 0]
    for num in nums:
        counts[num] += 1

    result = [0] * n
    red_index = white_index = 0
    for i in range(n):
        if counts[i] > 0:
            result[red_index] = i
            counts[i] -= 1
            red_index += 1
        else:
            while counts[result[white_index]] == 0 and white_index < red_index:
                white_index += 1
            if white_index < red_index:
                result[white_index] = i
                white_index += 1
            else:
                while counts[result[blue_index]] == 0 and blue_index < white_index:
                    blue_index += 1
                if blue_index < white_index:
                    result[blue_index] = i
                    blue_index += 1

    return result