148. 颜色分类（三指针）———荷兰国旗问题

给定一个包含红，白，蓝且长度为 n 的数组，将数组元素进行分类使相同颜色的元素相邻，并按照红、白、蓝的顺序进行排序。

我们可以使用整数 0，1 和 2 分别代表红，白，蓝。

 

 注意事项

不能使用代码库中的排序函数来解决这个问题。
排序需要在原数组中进行。

样例

给你数组 [1, 0, 1, 2], 需要将该数组原地排序为 [0, 1, 1, 2]。

思想：

 

设置两个指针left和right，分别指向第一个位置  和最后一个位置 ，然后设置一个遍历指针mid，从头开始进行遍历。

（1）若遍历到的位置为1，则说明它一定属于中部，根据总思路，中部的我们都不动，然后mid向前移动一个位置。

（2）若遍历到的位置为0，则说明它一定属于前部，于是就和left位置进行交换，然后mid向前移动一个位置，left也向前移动一个位置（表示前边的已经都排好了）。

（3）若遍历到的位置为2，则说明它一定属于后部，于是就和right位置进行交换，由于交换完毕后mid指向的可能是属于前部的，若此时mid前进则会导致该位置不能被交换到前部，所以此时mid不前进。而同1），right向前移动一个位置。

代码：

 

class Solution {
public:
    /*
     * @param nums: A list of integer which is 0, 1 or 2 
     * @return: nothing
     */
    void sortColors(vector<int> &nums) {
        // write your code here
   int left = 0, right = nums.size() - 1;
        int middle = 0;
        
        // should be <= not <
        while (middle <= right) {
            if (nums[middle] == 0) {
                swap(nums[middle], nums[left]);
                left++;
                middle++;
            } else if (nums[middle] == 1) {
                middle++;
            } else {
                swap(nums[middle], nums[right]);
                right--;
            }
        }
    }
};

思路二：

 

 

【思路二】

 

（1）遍历数组，统计红白蓝三色球（0，1，2）的个数

（2）根据红白蓝三色球（0，1，2）的个数重排数组

时间复杂度：O(n)

【代码】

 

[cpp] view plain copy

 

1. class Solution {  
2. public:  
3.     void sortColors(int A[], int n) {  
4.         if(n <= 1){  
5.             return;  
6.         }//if  
7.         // 统计个数  
8.         int red = 0,white = 0,blue = 0;  
9.         for(int i = 0;i < n;++i){  
10.             if(A[i] == 0){  
11.                 ++red;  
12.             }//if  
13.             else if(A[i] == 1){  
14.                 ++white;  
15.             }//else  
16.             else{  
17.                 ++blue;  
18.             }//else  
19.         }//for  
20.         // 重新布局  
21.         for(int i = 0;i < n;++i){  
22.             if(red > 0){  
23.                 A[i] = 0;  
24.                 --red;  
25.             }//if  
26.             else if(white > 0){  
27.                 A[i] = 1;  
28.                 --white;  
29.             }//else  
30.             else{  
31.                 A[i] = 2;  
32.             }  
33.         }//for  
34.     }  
35. };  
36.