leetcode 75. Sort Colors

来源:互联网 发布:在线查询域名被墙 编辑:程序博客网 时间:2024/05/17 09:13

这里写图片描述

 这是一道简单的排序题。由于整个数组只有0,1,2三种元素,所以计数排序是一个很好的解决方法,只要开一个大小为3的计数数组,遍历要排序的数组一次,用对应下标的计数数组元素的数值统计元素在数组中出现的次数,然后再经过一趟,将计数数组中的元素赋值给排序数组,就得到了最终的结果。计数排序可以认为是一种特殊的桶排序,其时间复杂度O(n)突破了一般排序的O(nlogn)下限,当元素的范围在0-100的时候使用起来非常方便。不过这题使用计数排序并不是最佳答案,但是讨论区中的代码我也一一尝试过,效率并没有明显提升。

 以下是计数排序的实现:

class Solution {public:    void sortColors(vector<int>& nums) {        vector<int> vec(3, 0);        for (int i = 0; i < nums.size(); i++) {            vec[nums[i]]++;        }        int index = 0;        for (int i = 0; i < vec.size(); i++) {            for (int j = 0; j < vec[i]; j++) {                nums[index] = i;                index++;            }        }    }};
原创粉丝点击