[leetcode]顔色分類

75. 顔色分類

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

此題中，我們使用整數 0、 1 和 2 分别表示紅色、白色和藍色。

注意:

不能使用代碼庫中的排序函數來解決這道題。

示例:

輸入: [2,0,2,1,1,0]
輸出: [0,0,1,1,2,2]
           

進階：

• 一個直覺的解決方案是使用計數排序的兩趟掃描算法。 首先，疊代計算出0、1 和 2 元素的個數，然後按照0、1、2的排序，重寫目前數組。
• 你能想出一個僅使用常數空間的一趟掃描算法嗎？

C++解法：

class Solution {
public:
    void sortColors(vector<int>& nums) {
        bool flag = true;
        for (int i = 0, length = nums.size(); i < length - 1 && flag; ++i)
        {
            flag = false;
            for (int j = length - 2; j >= i;--j)
            {
                if (nums[j] > nums[j+1])
                {
                    swap(nums[j],nums[j+1]);
                    flag = true;
                }
            }
        }
    }
};