【Leet Code】 75．Sort Colors--medium

最新推荐文章于 2020-03-23 14:48:14 发布

lili616

最新推荐文章于 2020-03-23 14:48:14 发布

阅读量335

点赞数

分类专栏： Leet Code -- Sort 文章标签： Data Struct arithmet

本文链接：https://blog.csdn.net/lili616/article/details/48469067

版权

Leet Code -- Sort 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种使用map结构对含有三种颜色元素的数组进行排序的方法，通过统计颜色数量并重新填充数组，实现了颜色相邻且按顺序排列的效果。

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

Given anarray with n objects colored red, white or blue, sort them so thatobjects of the same color are adjacent, with the colors in the order red, whiteand 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.

思路：

因为只有3种颜色，所以借助map结构，统计每种颜色的数量；

清空原来vector里面的内容；

按照map结构中统计的数据重新往vector里面加值。

class Solution {
public:
    void sortColors(vector<int>& nums) {
        map<int, int> counts;
        counts.insert({0, 0});
        counts.insert({1, 0});
        counts.insert({2, 0});
        
        for(auto item: nums)
        {
            switch(item)
            {
            case 0:
                ++counts[0];
                break;
            case 1:
                ++counts[1];
                break;
            case 2:
                ++counts[2];
                break;
            }
        }
        
        nums.clear();
        
        for(auto iter = counts.begin(); iter != counts.end(); ++ iter)
        {
            while(iter->second--)
                nums.push_back(iter->first);
        }
    
        return;
    }
};