75. 颜色分类


文章目录


1.题目

75. 颜色分类

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

我们使用整数 012 分别表示红色、白色和蓝色。

必须在不使用库内置的 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`

2.思路

  1. 定义两个指针
    • left:指向数组的最左边,表示 0(红色)的边界。
    • right:指向数组的最右边,表示 2(蓝色)的边界。
  2. 遍历数组
    • 如果当前数字是 0,就把它放到 left 的位置,然后 left 向右移动。
    • 如果当前数字是 2,就把它放到 right 的位置,然后 right 向左移动。
    • 如果当前数字是 1,就跳过,继续遍历。

注意:left 初始化为-1是因为最开始没有安排好第一个位置,所以不好初始化为0,right同理

3.代码

c++ 复制代码
class Solution {
public:
    void sortColors(vector<int>& nums) {
        // 定义两个指针,left 指向 0 的边界,right 指向 2 的边界
        int left = -1, right = nums.size();
        // tmp 是当前遍历的位置
        int tmp = 0;

        // 遍历数组
        while (tmp < right) {
            if (nums[tmp] == 0) {
                // 如果当前数字是 0,把它放到 left 的位置
                ++left;  // left 向右移动
                swap(nums[tmp], nums[left]);  // 交换当前数字和 left 位置的数字
                ++tmp;  // 继续遍历下一个数字
            } else if (nums[tmp] == 2) {
                // 如果当前数字是 2,把它放到 right 的位置
                --right;  // right 向左移动
                swap(nums[tmp], nums[right]);  // 交换当前数字和 right 位置的数字
                // 注意:这里不需要移动 tmp,因为交换后的数字可能是 0 或 1,需要再次检查
            } else {
                // 如果当前数字是 1,直接跳过
                ++tmp;  // 继续遍历下一个数字
            }
        }
    }
};

相关推荐
ysa0510309 分钟前
Dijkstra 算法#图论
数据结构·算法·图论
一定要AK39 分钟前
2025—暑期训练一
算法
一定要AK43 分钟前
贪心专题练习
算法
森焱森1 小时前
无人机三轴稳定控制(2)____根据目标俯仰角,实现俯仰稳定化控制,计算出升降舵输出
c语言·单片机·算法·架构·无人机
ytttr8732 小时前
matlab通过Q学习算法解决房间路径规划问题
学习·算法·matlab
go54631584653 小时前
修改Spatial-MLLM项目,使其专注于无人机航拍视频的空间理解
人工智能·算法·机器学习·架构·音视频·无人机
油泼辣子多加3 小时前
【Torch】nn.BatchNorm1d算法详解
算法
nlog3n3 小时前
基于 govaluate 的监控系统中,如何设计灵活可扩展的自定义表达式函数体系
算法·go
IT古董4 小时前
【第三章:神经网络原理详解与Pytorch入门】01.神经网络算法理论详解与实践-(2)神经网络整体结构
pytorch·神经网络·算法
ThetaarSofVenice4 小时前
垃圾收集相关算法Test
java·jvm·算法