头图

颜色分类

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

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

示例说明请见LeetCode官网。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/probl...
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法一:遍历数组

首先,声明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 + " ");
        }
    }
}
【每日寄语】 智者贵于乘时,时不可失。

醉舞经阁
1.8k 声望7.1k 粉丝

玉树临风,仙姿佚貌!