H100047. 颜色分类 ACM 版

中等

通过率 0% · 提交 0 · 通过 0

Hot100 ACM 开放训练 · 配套 LeetCode 75 图解动画 · 标准输入输出练习看动画讲解

给定长度为 n 的数组 nums，元素只可能是 0、1、2，分别表示三种颜色。请按 0、1、2 的顺序原地排序并输出结果。

这题属于 Hot100 ACM 开放训练中的「原地排序」方向。建议先看动画确认核心思路，再回到右侧编辑器按标准输入输出写完整代码。

第一行输入整数 n。第二行输入 n 个整数，表示 nums。

输出排序后的数组。

示例 1

输入示例

6
2 0 2 1 1 0

输出示例

0 0 1 1 2 2

说明：所有 0 在前，2 在后。

时间限制 2000 ms · 内存限制 256 MB

字号13Ctrl+/ 注释 · Ctrl+↵ 提交

import sys

def solve():
    data = sys.stdin.read().strip().split()
    # TODO: 按题目输入格式解析 data，并输出答案

if __name__ == "__main__":
    solve()

写完代码点「提交」，将对全部测试用例判题。