力扣第 75 题是 颜色分类

题目描述

输入 :一个数组 nums,其中元素只能是 012

输出:对数组进行排序,使其按顺序排列。

你需要使用 原地算法,即不能使用额外的空间来完成排序。


示例

示例 1

输入

plaintext 复制代码
nums = [2, 0, 2, 1, 1, 0]

输出

plaintext 复制代码
[0, 0, 1, 1, 2, 2]
示例 2

输入

plaintext 复制代码
nums = [2, 0, 1]

输出

plaintext 复制代码
[0, 1, 2]

解题思路

1. 计数排序
  • 遍历数组,统计 012 的个数。
  • 然后按照统计的数量依次将 012 填回数组。

这种方法需要两次遍历,时间复杂度为 O(n) ,空间复杂度为 O(1)


2. 双指针法

使用三个指针:

  • p0:指向下一个应该放置 0 的位置。
  • p2:指向下一个应该放置 2 的位置。
  • curr:用于遍历数组。

算法过程

  1. 初始化 p0 = 0curr = 0p2 = n - 1
  2. 遍历数组:
    • 如果 nums[curr] == 0:将 nums[curr]nums[p0] 交换,p0++curr++
    • 如果 nums[curr] == 2:将 nums[curr]nums[p2] 交换,p2--,但此时 curr 不前进(因为交换过来的元素还需要检查)。
    • 如果 nums[curr] == 1curr++

这种方法只需一次遍历,时间复杂度为 O(n) ,空间复杂度为 O(1)


实现代码

双指针法实现
c 复制代码
#include <stdio.h>

void sortColors(int* nums, int numsSize) {
    int p0 = 0, curr = 0, p2 = numsSize - 1;

    while (curr <= p2) {
        if (nums[curr] == 0) {
            // 交换 nums[curr] 和 nums[p0]
            int temp = nums[curr];
            nums[curr] = nums[p0];
            nums[p0] = temp;
            p0++;
            curr++;
        } else if (nums[curr] == 2) {
            // 交换 nums[curr] 和 nums[p2]
            int temp = nums[curr];
            nums[curr] = nums[p2];
            nums[p2] = temp;
            p2--;
        } else {
            curr++;
        }
    }
}

int main() {
    int nums[] = {2, 0, 2, 1, 1, 0};
    int numsSize = sizeof(nums) / sizeof(nums[0]);

    sortColors(nums, numsSize);

    // 输出排序后的数组
    for (int i = 0; i < numsSize; i++) {
        printf("%d ", nums[i]);
    }
    printf("\n");

    return 0;
}

代码解析

  1. 变量初始化

    • p0 指向当前应该放 0 的位置。
    • p2 指向当前应该放 2 的位置。
    • curr 遍历数组。
  2. 循环逻辑

    • 如果 nums[curr] == 0
      • 交换 nums[curr]nums[p0]
      • 更新 p0curr
    • 如果 nums[curr] == 2
      • 交换 nums[curr]nums[p2]
      • 更新 p2
    • 如果 nums[curr] == 1
      • 直接更新 curr
  3. 结束条件

    • curr > p2 时,所有元素已经按要求排序。

时间复杂度和空间复杂度

  • 时间复杂度

    • 只需一次遍历,时间复杂度为 O(n)
  • 空间复杂度

    • 没有使用额外的空间,空间复杂度为 O(1)
相关推荐
opple664 分钟前
力扣-数据结构-二叉树
数据结构·算法·leetcode
import_random27 分钟前
[社交网络]布局算法(可视化)
算法
时光话1 小时前
Lua 第14部分 数据结构
开发语言·数据结构·lua
地平线开发者1 小时前
C++ 部署的性能优化方法
c++·算法·自动驾驶
怀念无所不能的你1 小时前
acwing背包问题求方案数
学习·算法·动态规划·dp
Yingye Zhu(HPXXZYY)1 小时前
洛谷P12238 [蓝桥杯 2023 国 Java A] 单词分类
c++·算法·蓝桥杯
Phoebe鑫2 小时前
数据结构每日一题day13(链表)★★★★★
数据结构·链表
Seven972 小时前
缓存穿透的解决方式?—布隆过滤器
java·数据结构·redis
积极向上的向日葵2 小时前
链表的中间节点
数据结构·算法·链表·快慢指针
曾几何时`2 小时前
C++——哈希表
算法·哈希算法