leetcode--Sort Colors-优快云博客

本文链接：https://blog.youkuaiyun.com/sinat_24520925/article/details/45790487

本文介绍了一种特殊的排序问题——三色排序，即给定一个包含红、白、蓝三种颜色的数组，要求按红、白、蓝的顺序进行排序。讨论了两种不同的实现方法：一种是使用额外的空间达到线性时间复杂度；另一种是在原地操作，减少空间复杂度。

题目描述：

Given an array with n objects colored red, white or blue, sort them so that objects of the same color are adjacent, with the colors in the order red, white and blue.

Here, we will use the integers 0, 1, and 2 to represent the color red, white, and blue respectively.

Note:
You are not suppose to use the library's sort function for this problem.

解题思路

求只有三个值的数组（0，1，2）的排序。

方法一

重新开辟一段内存，先将0存入，再存入1，之后存入2.时间复杂度为O（n）空间复杂度为O（n）代码如下：

class Solution {
public:
    void sortColors(vector<int>& nums) {
        if(nums.size()==1) return;
        vector<int> aa;
        for(int col=0;col<3;col++)
        {
            for(int i=0;i<nums.size();i++)
        {
            if(nums[i]==col)
            aa.push_back(nums[i]);
        }
        }
        nums=aa;
    }
};

方法二

从数组前后向中间查找，0的话跟前面数组交换，1的话跟后面节点交换，时间复杂度为O（n/2），空间复杂度为O（1）

class Solution {
public:
    void sortColors(vector<int>& nums) {
        if(nums.size()==1) return;
        vector<int> aa;
        for(int col=0;col<3;col++)
        {
            for(int i=0;i<nums.size();i++)
        {
            if(nums[i]==col)
            aa.push_back(nums[i]);
        }
        }
        nums=aa;
    }
};