leetcode 颜色分类

最新推荐文章于 2024-07-13 22:53:27 发布

zhengqiuQAQ

最新推荐文章于 2024-07-13 22:53:27 发布

阅读量121

点赞数 1

分类专栏： LEETCODE 文章标签：快速排序 c++ 数据结构

本文链接：https://blog.csdn.net/qq_22684941/article/details/119566109

版权

LEETCODE 专栏收录该内容

87 篇文章 0 订阅

订阅专栏

该博客介绍了一种使用双指针在O(n)时间和O(1)空间复杂度内对包含红色、白色和蓝色元素的数组进行排序的方法。通过设置头尾指针，遍历数组，将0与头指针位置交换，2与尾指针位置交换，实现原地排序，使得数组按0、1、2的顺序排列。

摘要由CSDN通过智能技术生成

65.颜色分类

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

此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

示例 1：
输入：nums = [2,0,2,1,1,0]
输出：[0,0,1,1,2,2]

示例 2：
输入：nums = [2,0,1]
输出：[0,1,2]

示例 3：
输入：nums = [0]
输出：[0]

示例 4：
输入：nums = [1]
输出：[1]

提示：

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

进阶：

你可以不使用代码库中的排序函数来解决这道题吗？
你能想出一个仅使用常数空间的一趟扫描算法吗？

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sort-colors。

直接使用标准库排序
复杂度根据使用的库而定。

void sortColors(vector<int>& nums) {
        // 标准库
        std::sort(nums.begin(), nums.end());
    }

进阶：双指针
采用双指针遍历一次，时间复杂度o(n),空间复杂度o(1).
设置头尾指针，遍历一次数组，等于0与头指针位置交换，头指针加1；等于2与尾指针位置交换，尾指针减1.

以示例1为例：

次数	value： 2,0,2,1,1,0
第一次	0,0,2,1,1,2
第二次	0,0,2,1,1,2
第三次	0,0,1,1,2,2
第四次	0,0,1,1,2,2
条件中止

void sortColors(vector<int>& nums) {
        // 双指针
        int p_start = 0, p_end = nums.size() - 1;
        for (int i = 0; i < nums.size(); ++i) {
            while (nums[i] == 2 && i < p_end) {
                std::swap(nums[i], nums[p_end]);
                --p_end;
            }
            if (nums[i] == 0) {
                std::swap(nums[i], nums[p_start]);
                ++p_start;
            }
        }
    }