![这里写图片描述](https://img-blog.csdn.net/20171115212714492?watermark/2/text/aHR0cDovL2Jsb2cuY3Nkbi5uZXQvcXFfMzg3Njc2Njg=/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70/gravity/SouthEast)
这是一道简单的排序题。由于整个数组只有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++;
}
}
}
};