Leetcode75. 颜色分类

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

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

输入：nums = [2,0,2,1,1,0] 输出：[0,0,1,1,2,2]

示例 2：

输入：nums = [2,0,1] 输出：[0,1,2]

示例 3：

输入：nums = [0] 输出：[0]

示例 4：

输入：nums = [1] 输出：[1]

做题思路

这个题目我们使用 Arrays.sort() 解决，哈哈，但是那样太无聊啦，题目含义就是让我们将所有的 0 放在前面，2放在后面，1 放在中间，是不是和我们上面说的荷兰国旗问题一样。我们仅仅将 1 做为 pivot 值。

public class lc75 {
    public int[] sortColors(int[] nums) {
        int len = nums.length;
        int left = 0;

        int right = len-1;
        //<=right为标准，不用从头扫到尾
       for (int i =0;i<=right ;i++){
           if(nums[i]==0){
               swap(nums,i,left);
               left++;
           }
           if(nums[i]==2){
               swap(nums,i,right);
               right--;
               //如果不等于 1 则需要继续判断，所以不移动 i 指针，i--
               if (nums[i]!=1){
                   i--;
               }
           }
       }
        return nums;
    }
    public void swap(int[] nums,int a, int b){
        int temp = nums[a];
        nums[a]= nums[b];
        nums[b] = temp;
    }
}