【力扣-数据结构基础】【第 2 天】75. 颜色分类

标题	75. 颜色分类
难度	中等
天数	第2天,第1/3题
数据结构	数组

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

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

必须在不使用库的sort函数的情况下解决这个问题。

示例 1：

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

示例 2：

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

提示：

• n == nums.length
• 1 <= n <= 300
• nums[i] 为 0、1 或 2

进阶：

• 你可以不使用代码库中的排序函数来解决这道题吗？
• 你能想出一个仅使用常数空间的一趟扫描算法吗？

以上内容来源：力扣（LeetCode）
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

1. 只针对红(0) 蓝(2)进行移动，红的全部移动到左边，蓝色全部移动到右边
2. 采用双指针操作，
   • left代表左边可放入红色的位置
   • right代表右边可放入蓝色的位置
   • index 代表遍历的位置

	int m = nums.length;
	//指向左边元素
	int left = 0;
	//指向右边元素
	int right = m - 1;
	//当前遍历位置
	int index = 0;

3. while循环条件 index < m
4. 遇到0左移到left，小优化，如果index > left才移动
   • 然后左指针右移一位

	while(index < m){
		int item = nums[index];
		if(item == 0 && index > left){
		swap(index,left,nums);
		left++;
	}

5. 交换方法 swap

	public void swap(int index,int toIndex,int[] nums){
        int temp = nums[toIndex];
        nums[toIndex] = nums[index];
        nums[index] = temp;
    }

6. 遇到2右移到right

 	while(index < m){
 		if(item == 2 && index < right){
			swap(index,right,nums);
			right--;
		}
	}

7. 如果遇到1那么没有移动操作，直接遍历下一个元素

完整代码

class Solution {
    //数据结构基础  第 2 天  数组 1/3
    public void sortColors(int[] nums) {
        int m = nums.length;
        //指向左边元素
        int left = 0;
        //指向右边元素
        int right = m - 1;
        //当前遍历位置
        int index = 0;
        while(index < m){
            int item = nums[index];
            if(item == 0 && index > left){
                swap(index,left,nums);
                left++;
            }else if(item == 2 && index < right){
               swap(index,right,nums);
               right--;
            }else{
                index++;
            }
        }
    }
    public void swap(int index,int toIndex,int[] nums){
        int temp = nums[toIndex];
        nums[toIndex] = nums[index];
        nums[index] = temp;
    }
}