Leetcode--Java--75. 颜色分类

写点code吧

于 2022-04-08 20:19:46 发布

阅读量862

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode

本文链接：https://blog.youkuaiyun.com/Sherlock_Obama/article/details/124049898

Leetcode 专栏收录该内容

395 篇文章

订阅专栏

这是一篇关于LeetCode第75题的解决方案，使用Java编程语言。博客详细介绍了如何对包含红、白、蓝三种颜色的数组进行原地排序，使其按颜色顺序排列，同时避免使用内置排序函数。文章通过荷兰国旗问题的思路，利用三指针技术进行划分和排序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

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

思路

荷兰国旗问题 + 三指针

用三个指针来划分区间，0在j的左边，j和i之间是1，k的右边是2，有待探索的是i到k之间

代码

class Solution {
    public void sortColors(int[] nums) {
        int i = 0, j = 0, k = nums.length - 1;
        while (i <= k) {
            if (nums[i] == 0) {
                swap(nums, i, j);
                i ++;
                j ++;
            } else if (nums[i] == 2) {
                swap(nums, k, i);
                k --;
            } else {
                i ++;
            }
        }
    }
    public void swap(int nums[], int i, int j) {
        int t = nums[i];
        nums[i] = nums[j];
        nums[j] = t;
    }
}