75.颜色分类

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

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

注意:
不能使用代码库中的排序函数来解决这道题。

示例:
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

进阶:
一个直观的解决方案是使用计数排序的两趟扫描算法。
首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗?

思路
1、采用冒泡排序

#include 
#include 
using namespace std;
class Solution {
public:
    void sortColors(vector& nums) {
        int blue = 0, red = 0, green = 0;
        for (int i = 0; i < nums.size(); i++)
        {
            if (nums[i] == 0) blue++;
            if (nums[i] == 1) red++;
            if (nums[i] == 2) green++;
        }
        for (int i = 0; i < blue; i++) nums[i] = 0;
        for (int i = blue; i < blue + red; i++) nums[i] = 1;
        for (int i = blue + red; i < nums.size(); i++) nums[i] = 2;
    }
};

int main(int argc, char* argv[])
{
    vector test = { 2,0,2,1,1,0, 1, 2};
    Solution().sortColors(test);
    return 0;
}

2、先统计三种颜色的个数,然后将数组对应的位置填数

    void sortColors(vector& nums) {
        int n = nums.size();
        for(int i=0; i < nums.size()-1; i++) {
            for(int j=0; j < n-1-i; j++) {
                if(nums[j] > nums[j+1]) {
                    int temp = nums[j+1];
                    nums[j+1] = nums[j];
                    nums[j] = temp;
                }
            }
        }
    }

你可能感兴趣的:(75.颜色分类)