LeetCode-075-颜色分类

颜色分类

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

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

示例说明请见LeetCode官网。

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/sort-colors/

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法一：遍历数组

首先，声明2个参数，zeroIndex用来记录从左开始第一个不为红色的位置，初始值为0，twoIndex用来记录从右开始第一个不为蓝色的位置，初始值为nums.length - 1，然后遍历数组，遍历过程如下：

• 如果当前的值等于2，则将当前位置的值和twoIndex位置的值进行交换，然后将twoIndex减一，然后进行下一轮处理；
• 如果当前的值等于1，则继续判断下一个数字；
• 如果当前的值等于0，则将当前位置的值和zeroIndex位置的值进行交换，然后将zeroIndex加一，然后进行下一轮处理。
• 遍历结束的位置是twoIndex，因为twoIndex可能会左移，也就是不用遍历到数组的最后一位。

public class LeetCode_075 {
    public static void sortColors(int[] nums) {
        if (nums.length == 1) {
            return;
        }
        // zeroIndex用来记录从左开始第一个不为红色的位置，twoIndex用来记录从右开始第一个不为蓝色的位置
        int zeroIndex = 0, twoIndex = nums.length - 1;
        for (int i = 0; i <= twoIndex; ) {
            if (nums[i] == 2) {
                // 如果当前的值等于2，则将当前位置的值和twoIndex位置的值进行交换
                nums[i] = nums[twoIndex];
                nums[twoIndex] = 2;
                twoIndex--;
                continue;
            } else if (nums[i] == 1) {
                i++;
            } else if (nums[i] == 0) {
                // 如果当前的值等于0，则将当前位置的值和zeroIndex位置的值进行交换
                nums[i] = nums[zeroIndex];
                nums[zeroIndex] = 0;
                zeroIndex++;
                i++;
            }
        }
    }

    public static void main(String[] args) {
        int[] nums = new int[]{2, 0, 2, 1, 1, 0};
        sortColors(nums);
        for (int num : nums) {
            System.out.print(num + " ");
        }
    }
}

【每日寄语】 智者贵于乘时，时不可失。