欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

CountSort

程序员文章站 2022-03-24 13:49:22
...

马士兵老师的算法课

3.计数排序

算法思想:
使用的限定范围为数据量大但范围小时 开辟两个数组,result负责记录结果,count负责记录每个index出现的个数

1.遍历整个数组,得到count计数数组;
2.count[i]=count[i-1]+count[i],得到累加数组,目的是记录该index的最后一个的下标;
3.从末尾向前遍历原数组,按照累加数组,将对应的数放到result的index-1上

时间复杂度:O(n+k) n为遍历原数组,k为遍历count
空间复杂度:O(n+k) n为开辟的result,k为开辟的count

代码:

class Countsort
{
public:
	
	vector<int> countsort(vector<int>& arr)
	{
		vector<int> result(arr.size());
		vector<int> count(int(10));
		for (int i = 0; i < arr.size(); i++) //count数组用来计数,每个下标值为index的有多少个
			count[arr[i]]++;
		for (int i = 1; i < count.size(); i++) //累加数组,用于记录每个index最后一个数据的位置
		{
			count[i] = count[i] + count[i - 1];
		}
		for (int i = arr.size() - 1; i >= 0; i--) //从原数组的最后一位开始遍历,
		{                                         //根据累加数组的信息,放在result数组对应的位置上,这样可以保证数组的稳定性
			result[--count[arr[i]]] = arr[i];
		}
		for (int i = 0; i < result.size(); i++)
			cout << result[i] << " ";
		return result;
	}
};
相关标签: 算法学习