75. 颜色分类

题目描述

给定一个包含红色、白色和蓝色，一共 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]

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sort-colors
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

• 双支针一前一后，从前往后遍历，用 if 就可以保证left前都是0，left - i 间都是1；
• 而当遇到2时，若用 if 交换后当前值仍有可能是2，再往后遍历就不能保证 left - i 间都是1，导致错误。用 while 时，则可保证当前值不为2，然后进行下面判断，为0进行 left 交换，为1不做操作。

class Solution {
    public void sortColors(int[] nums) {
        int n = nums.length;
        int left = 0,right = n-1;
        for(int i = 0;i <= right;++i){
            while(i < right && nums[i]==2){
                int tem1 = nums[i];
                nums[i] = nums[right];
                nums[right] = tem1;
                --right;
            }
            if(nums[i]==0){
                int tem2 = nums[i];
                nums[i] = nums[left];
                nums[left] = tem2;
                ++left;
            }
        }
    }
}