LeetCode 75 颜色分类

Ethan-running

于 2024-03-15 16:15:40 发布

阅读量386

点赞数 6

分类专栏： # leetcode 文章标签： leetcode 算法数据结构

本文链接：https://blog.youkuaiyun.com/qq_43745578/article/details/136743109

版权

文章介绍了如何在不使用内置排序函数的情况下，通过统计和原地交换整数数组中红色、白色和蓝色元素，实现按照0、1、2的顺序进行排序。给出了三种不同的解法，包括统计数量后重新赋值、单指针和双指针方法，所有算法的时间复杂度均为O(n)，空间复杂度为O(1)。

摘要生成于 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]

提示：

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

进阶：

你能想出一个仅使用常数空间的一趟扫描算法吗？

解法

解法1：先统计0、1、2的数量，再重新赋值

java代码：

class Solution {
   
    public void sortColors(int[] nums) {
   
        // 统计数量
        int num0 = 0, num1 = 0, num2 = 0;
        for (int num : nums) {
   
            if (num == 0) {
   
                num0 += 1;
            }
            if (num == 1) {
   
                num1 += 1;
            }
            if (num