颜色分类

最新推荐文章于 2021-02-03 13:56:40 发布

原创最新推荐文章于 2021-02-03 13:56:40 发布 · 733 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#lintcode #Lentitude #颜色分类

LintCode_数组专栏收录该内容

25 篇文章

订阅专栏

本文介绍了一种有效的原地排序算法，用于将包含红、白、蓝三种颜色的数组按顺序排列。通过使用0、1、2代表不同颜色，文章提供了一个具体的C++实现示例，避免使用标准库排序函数。

描述

给定一个包含红，白，蓝且长度为 n 的数组，将数组元素进行分类使相同颜色的元素相邻，并按照红、白、蓝的顺序进行排序。

我们可以使用整数 0，1 和 2 分别代表红，白，蓝。

注意事项

不能使用代码库中的排序函数来解决这个问题。
排序需要在原数组中进行。

样例

给你数组 [1, 0, 1, 2], 需要将该数组原地排序为 [0, 1, 1, 2]。

思考

各种排序
也可以计算各个的数量，再 push_back()

代码

//  By Lentitude


class Solution{
public:
    /**
     * @param nums: A list of integer which is 0, 1 or 2 
     * @return: nothing
     */    
    void sortColors(vector<int> &nums) {
        // write your code here
        if (nums.empty()) return;

        bool isExist;
        for (int i = 0; i != nums.size(); ++i){
            isExist = false;
            for (int j = 1; j != nums.size() - i; ++j){
                if (nums[j - 1] > nums[j] ){
                    swap(nums[j - 1], nums[j]);
                    isExist = true;
                }
            }

            if (!isExist){
                return;
            }
        }
    }
};