对0、1和2的数组进行排序(简单计数)

这是一道关于排序算法的题目，给定一个只包含0、1和2的数组，将其排序，使得0在前，1在中间，2在末尾。为了解决这个问题，我们可以采用简单计数排序的思想，先扫描一遍数组，记录0、1、2每个数出现的次数，然后再按顺序输出数组中的元素。接下来我们就可以根据这个思路编写代码了。

代码

def sortColors(nums: List[int]) -> None:
    """
    Do not return anything, modify nums in-place instead.
    """
    count = [0, 0, 0]
    for num in nums:
        count[num] += 1
    i = 0
    for j in range(3):
        while count[j] > 0:
            nums[i] = j
            i += 1
            count[j] -= 1

代码解释

首先我们定义一个长度为3的数组count，用来记录每个数字出现的次数。然后我们遍历整个数组nums，将每个数字在count数组中对应的位置上加1。这样我们就可以得到数字0、1、2在数组中分别出现的次数。

接着我们定义一个变量i，用来记录当前已经输出了多少个数字。接下来我们遍历count数组，每遍历到一个数字j，就将数字j输出count[j]次，即按照数字出现的次数依次输出。

最终输出的结果就是排序好的数组，其中0在前，1在中间，2在末尾。

时间复杂度与空间复杂度

这个算法只需要遍历一遍输入数组，和遍历一遍长度为3的计数数组，因此时间复杂度是O(n)，其中n是数组的长度。另外，我们还需要一个长度为3的计数数组，因此空间复杂度也是O(1)。

总结

本题是一道经典的排序算法题目，主要考查了计数排序的思想。虽然这个算法时间复杂度只有O(n)，但是在某些情况下并不优于快速排序，在实际工程应用中需要谨慎使用。