- 找出待排序的数组中最⼤和最⼩的元素;
2. 统计数组中每个值为 i 的元素出现的次数,存⼊数组 C 的第 i 项;
3. 对所有的计数累加(从 C 中的第⼀个元素开始,每⼀项和前⼀项相加);
4. 向填充⽬标数组:将每个元素 i 放在新数组的第 C[i] 项,每放⼀个元素就将 C[i] 减去 1;
代码实现
//countSort.h
#include <vector>
#include <algorithm>
using std::vector;
void count_sort(vector<int>&);
//countSort.cc
#include "countSort.h"
void count_sort(vector<int>& nums) {
//确保排序容器非空
if(nums.size() == 0) {
return;
}
//使用nums的最大值 + 1 作为计数容器count_vec的大小
//+ 1 目的是最后一个位置存储的是前面所有键值的和
size_t count_vec_len = (*max_element(begin(nums), end(nums))) + 1;
//max_element()是用来来查询最大值所在的第一个位置
//*max_element() 表示的是最大值
vector<int> count_vec(count_vec_len, 0);
//统计每个键值出现的次数
for(int i = 0; i < nums.size(); ++i) {
count_vec[nums[i]]++;
}
#if 0
//后面的键值出现的位置为前面所有键值出现的次数之和
for(int i = 1; i < count_vec_len; ++i) {
}
#endif
for(int i = 0, k = 0; i < count_vec_len; ++i) {
//i用来遍历count_vec容器
//k用来重新找nums容器
for(int j = 0; j < count_vec[i]; ++j) {
//j用来循环放回
nums[k] = i;
++k;
}
}
}
max_element
是用来来查询最大值所在的第一个位置
//main.cc
#include <iostream>
#include "selectSort.h"
#include <cstdlib>
#include <ctime>
#include "heapSort.h"
#include "mergeSort.h"
#include "countSort.h"
using std::cout;
using std::endl;
const size_t nums_len = 10;
void print(vector<int>& nums) {
for(auto i : nums) {
cout << i << ' ';
}
cout << endl;
}
void test() {
//创建测试数据
std::srand(std::time(nullptr));
vector<int> nums;
for(int i = 0; i < nums_len; ++i) {
nums.push_back(std::rand() % 99);
}
print(nums);
//select_sort(nums);
//heap_sort(nums);
//merge_sort(nums, 0, nums_len);
count_sort(nums);
print(nums);
}
int main(void)
{
test();
}
用途和限制
- 面试题中出现“最快”、时间要求敏感
- 数据规模可控
- 考虑 数据重复
- 浮点数不会考虑重复,考虑重复说明是整形数据
限制:需要申请额外的空间