Leetcode Hot100 75.颜色分类

挺简单的一个问题，但太久没做题，导致编码花了不少时间（有点菜）。自己想到了两种思路，第二种写得有点复杂。

第一种思路

扫描一遍统计出数组中0,1,2的数量，然后重写数组。

第二种思路

两个指针。
左指针指向从左到右第一个不为0的数字，右指针指向从右到左第一个不为2的数字。
我们便可以遍历左右指针中间的数字，看是否有0或2，若有，则将它们放到该放的位置。（一次循环只找一个，效率较低）

class Solution(object):
    def sortColors(self, nums):
        left = 0
        right = len(nums) - 1
        while left < right - 1:
            while left < len(nums) and nums[left] == 0:
                left += 1
            while right > 0 and nums[right] == 2:
                right -= 1
            flag = 0
            for i in range(left, right + 1):
                if nums[i] == 2:
                    tmp = nums[right]
                    nums[right] = nums[i]
                    nums[i] = tmp
                    flag = 1
                    break

                elif nums[i] == 0:
                    tmp = nums[left]
                    nums[left] = nums[i]
                    nums[i] = tmp
                    flag = 1
                    break

            if flag == 0:
                break
        return nums

第三种思路

是LeetCode官方题解介绍的，挺简单，循环两次，一次把0放到数组前面，第二次把2放到后面。复杂度比第二种思路低。

第四种思路

比较妙的双指针，一个指向下一个0交换过来应该到的位置，一个指向下一个1交换过来应该到的位置，从左往右。详情可见。