題目

給定一個包含紅色、白色和藍色，一共 n 個元素的數(shù)組，原地對它們進行排序，使得相同顏色的元素相鄰，并按照紅色、白色、藍色順序排列。

此題中，我們使用整數(shù) 0、 1 和 2 分別表示紅色、白色和藍色。

注意:
不能使用代碼庫中的排序函數(shù)來解決這道題。

示例:

輸入: [2,0,2,1,1,0]
輸出: [0,0,1,1,2,2]
進階：

一個直觀的解決方案是使用計數(shù)排序的兩趟掃描算法。
首先，迭代計算出0、1 和 2 元素的個數(shù)，然后按照0、1、2的排序，重寫當前數(shù)組。
你能想出一個僅使用常數(shù)空間的一趟掃描算法嗎？

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/sort-colors
著作權歸領扣網(wǎng)絡所有。商業(yè)轉載請聯(lián)系官方授權，非商業(yè)轉載請注明出處。

解題思路

使用三個指針分別保存0，1， 2的位置。
p0從左向右遞增，p2從右向左遞減。
從頭掃描：
== 0: 交換p0和p1的值，遞增p0，遞增p1
== 1: 遞增p1
== 2: 交換p1和p2， 因為交換的值可能是1或者0，所以這里不能遞增p1，只能遞減p2.

代碼

class Solution(object):
    def sortColors(self, nums):
        """
        :type nums: List[int]
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        p0 = 0
        p1 = 0
        p2 = len(nums) - 1
        
        while p1 <= p2:
            if nums[p1] == 0:
                nums[p0], nums[p1] = nums[p1], nums[p0]
                p0 += 1
                p1 += 1
            elif nums[p1] == 2:
                nums[p2], nums[p1] = nums[p1], nums[p2]
                p2 -= 1
                # p1 不能前進
                # p1 += 1
            else:
                p1 += 1