力扣71. 颜色分类

颜色分类

题目

给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

必须在不使用库内置的 sort 函数的情况下解决这个问题。

思路

利用指针的思想，现在是有0，1，2让他们进行排序。首先，用i指针来控制遍历的位置，k指针指向小于1的数，j指针指向大于1的数。这样就使得1左边的都是小于1的，1右边的都是大于1的。

def color_classification(nums):
    k, i, j = 0, 0, len(nums)-1
    while i <=j:
        if nums [i] < 1:
            nums[k], nums[i] = nums[i], nums[k]
            i = i+1
            k = k + 1
        elif nums[i] > 1:
            nums[j], nums[i] = nums[i], nums[j]
            j = j - 1
        else:
            i = i+1
    return nums