No.75 Sort Colors

No.75 Sort Colors

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.

解法：

　　颜色排序：数组nums中包括代表红白蓝三色的0/1/2的n个数，
　　输出：经排序后，nums是有序的红白蓝即0/1/2有序排列
　　注意：0、1、2连续，可为数组下标，更简洁

class Solution
{
public:
    void sortColors(vector<int> &nums)
    {//颜色排序：数组nums中包括代表红白蓝三色的0/1/2的n个数，
     //输出：经排序后，nums是有序的红白蓝即0/1/2有序排列
     //注意：0、1、2连续，可为数组下标，更简洁
        int count = nums.size();
        if(count <=1)
            return;
        int color[3] = {0};//颜色统计数组
        
        for(int i=0; i<count; i++)
            color[nums[i]]++;//技巧

        for(int i=0,index=0; i<3; i++)
            for(int j=0; j<color[i]; j++)
                nums[index++] = i;
        //so简洁，so easy
    }
};

我的笨代码：

#include "stdafx.h"
#include <vector>
#include <iostream>
using namespace std;

class Solution
{
public:
    void sortColors(vector<int> &nums)
    {//颜色排序：数组nums中包括代表红白蓝三色的0/1/2的n个数，
     //输出：经排序后，nums是有序的红白蓝即0/1/2有序排列
        int count = nums.size();
        if(count <= 1)
            return;

        //类似计数排序
        int countRed = 0;
        int countWhite = 0;
        int countBlue = 0;
        //从前向后遍历
        for(int i=0; i<count; i++)
        {
            if(nums[i] == 0)
                countRed++;
            else if(nums[i] == 1)
                countWhite++;
            else if(nums[i] == 2)
                countBlue++;
        }

        //根据统计结果，从前向后改变数组
        for(int i=0; i<count; i++)
        {
            if(countRed != 0)
            {
                nums[i] = 0;
                countRed--;
                continue;
            }
            if(countWhite != 0)
            {
                nums[i] = 1;
                countWhite--;
                continue;
            }
            if(countBlue != 0)
            {
                nums[i] = 2;
                countBlue--;
                continue;
            }
        }
    }     
};
int main()
{
    Solution sol;
    int data[] = {0,2,1,2,1,2,0,0,0,1};
    vector<int> test(data,data+sizeof(data)/sizeof(int));

    for(auto &i : test)
        cout << i << " ";
    cout << endl;
    sol.sortColors(test);
        for(auto &i : test)
        cout << i << " ";
    cout << endl;

    return 0;
}