Day40 颜色分类-CFANZ编程社区

Day40 颜色分类

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

https://leetcode-cn.com/problems/sort-colors/

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

示例1：

示例2：

示例3：

示例4：

提示：

Java解法

package sj.shimmer.algorithm.m2;

import sj.shimmer.algorithm.Utils;

/**
 * Created by SJ on 2021/3/5.
 */

class D40 {
    public static void main(String[] args) {
//        int[] nums = {2, 0, 2, 1, 1, 0};
//        int[] nums = {2,0,1};
        int[] nums = {0};
        sortColors(nums);
        Utils.logArray(nums);
    }

    public static void sortColors(int[] nums) {
        if (nums == null) {
            return;
        }
        int length = nums.length;
        int left = -1;
        int right = length;
        int index = 0;
        int temp = 0;
        while (index < length) {
            if (nums[index] == 0&&index>left) {
                left++;
                temp = nums[left];
                nums[left] = nums[index];
                nums[index] = temp;
            } else if (nums[index] == 2&&index<right) {
                right--;
                temp = nums[right];
                nums[right] = nums[index];
                nums[index] = temp;
            } else {
                index++;
            }
        }
    }
}