每日一算法

1.红黄蓝球问题

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

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

注意:
不能使用代码库中的排序函数来解决这道题。

示例:

输入: [2,0,2,1,1,0]

输出: [0,0,1,1,2,2]

思路

结合三路快排partition思路的应用，设定两个索引，一个从左往右滑动zero，一个从右往左滑动two，遍历nums，当nums[i]的值为1时，i++；当nums[i]的值为2时，two的值先减1，而后交换nums[i]与nums[two]，此时在观察nums[i]的值；当nums[i]的值为0时，zero++，而后交换nums[i]与nums[zero]，i++;当 i = two时，结束循环。

# 代码实现，Python 3
def cal(x):
    zero = -1
    two = len(x)
    i = 0
    while i != two:
        if x[i] == 1:
            i += 1
        elif x[i] == 2:
            two -= 1
            x[i], x[two] = x[two], x[i]
        elif x[i] == 0:
            zero += 1
            x[i], x[zero] = x[zero], x[i]
            i += 1
    print(x)