leetcode75.颜色分类

题目大意

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

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

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

解题思路

荷兰国旗问题。
首先设置3个变量left，index和right。left表示0~left部分都是0， right表示right ~length部分都是2，index表示当前遍历的位置。三个变量的初始情况如下图所示。

• 若nums[index] == 2：则将该位置的值与nums[right-1]位置的值交换，并将right向左移动一位（因为right右边都是2，所以将index位置的2交换过来后，应该将其纳入right的范围，因此right-1），index指针不移动，继续在该位置进行判断（因为交换过来的值不确定）；
• 若nums[index] ==0：表示应该将该位置的值与num[left+1]位置进行交换，并将left向右移动一位（因为left范围内都是0，所以将index位置的0交换过来后，应该将其纳入left的范围内，因此left+1），index指针向后移动一位（因为left左边全部为0，left~index范围内的值全部为1，因此交换过来的值必定是1，不需要再进行判断）；
• 若nums[index]==1：不进行操作，index继续移动。

class Solution {
public:
    void sortColors(vector<int>& nums) {
        if (nums.empty())
        	return ;

        int left = -1, right = nums.size(), index = 0;

        while(index < right){
        	if (nums[index] == 0){
        		swap(nums[index++], nums[++left]);
        	}
        	else if (nums[index] == 2)
        		swap(nums[--right], nums[index]);
        	else{
        		++index;
        	}
        }
    }
};