数据结构——排序算法——计数排序

计数排序就是一种时间复杂度为 O(n) 的排序算法,该算法于 1954 年由 Harold H. Seward 提出。在对一定范围内的整数排序时,它的复杂度为 O(n+k)(其中 k 是整数的范围大小)。

伪计数排序

我们需要对一列数组排序,这个数组中每个元素都是 [1,9] 区间内的整数。那么我们可以构建一个长度为 9 的数组用于计数,计数数组的下标分别对应区间内的 9 个整数。然后遍历待排序的数组,将区间内每个整数出现的次数统计到计数数组中对应下标的位置。最后遍历计数数组,将每个元素输出,输出的次数就是对应位置记录的次数。

void countingSort9(std::vector<int>& arr) {
	// Create an array of length 9, where indices 0-8 correspond to numbers 1-9
	//建立长度为 9 的数组,下标 0~8 对应数字 1~9
	std::vector<int> counting(9, 0);
	// Iterate through each element in the input array
	//建立长度为 9 的数组,下标 0~8 对应数字 1~9
	for (int element : arr) {
		// Count the occurrences of each integer and store them in the counting array
		//建立长度为 9 的数组,下标 0~8 对应数字 1~9
		counting[element - 1]++;
	}
	int index = 0;
	// Iterate through the counting array and output each element
	//遍历计数数组,将每个元素输出
	for (int i = 0; i < 9; i++) {
		// Output each element as many times as it appears in the input
		//遍历计数数组,将每个元素输出
		while (counting[i] != 0) {
			arr[index++] = i + 1;
			counting[i]--;
		}
	}
}

算法非常简单,但这里的排序算法 并不是 真正的计数排序。因为现在的实现有一个非常大的弊端:排序完成后,arr 中记录的元素已经不再是最开始的那个元素了,他们只是值相等,但却不是同一个对象。

在纯数字排序中,这个弊端或许看起来无伤大雅,但在实际工作中,这样的排序算法几乎无法使用。因为被排序的对象往往都会携带其他的属性,但这份算法将被排序对象的其他属性都丢失了。

伪计数排序 2.0

对于这个问题,我们很容易想到一种解决方案:在统计元素出现的次数时,同时把真实的元素保存到列表中,输出时,从列表中取真实的元素。

void countingSort9(std::vector<int>& arr) {
	// Create an array of length 9, where indices 0-8 correspond to numbers 1-9
	// 建立长度为 9 的数组,下标 0~8 对应数字 1~9
	std::vector<int> counting(9, 0);
	// Record the actual elements at each index, using a queue to maintain stability
	//记录每个下标中包含的真实元素,使用队列可以保证排序的稳定性
	std::unordered_map<int, std::queue<int>> records;
	// Iterate through each element in the input array
	//遍历 arr 中的每个元素
	for (int element : arr) {
		// Count the occurrences of each integer and store them in the counting array
		// 将每个整数出现的次数统计到计数数组中对应下标的位置
		counting[element - 1]++;
		// Initialize a queue for each unique element if it doesn't exist
		//初始化每个唯一元素的队列(如果不纯在)
		if (records.find(element - 1) == records.end()) {
			records[element - 1] = std::queue<int>();
		}
		// Add the element to the corresponding queue
		//将元素添加到相应的队列
		records[element - 1].push(element);
	}
	int index = 0;
	// Iterate through the counting array and output each element
	// 遍历计数数组,将每个元素输出
	for (int i = 0; i < 9; i++) {
		// Output the element as many times as it appears in the input
		//输出的次数就是对应位置记录的次数
		while (counting[i] != 0) {
			// Output the actual element from the queue
			//输出的次数就是对应位置记录的次数
			arr[index++] = records[i].front();
			records[i].pop();
			counting[i]--;
		}
	}
}

在这份代码中,我们通过队列来保存真实的元素,计数完成后,将队列中真实的元素赋到 arr 列表中,这就解决了信息丢失的问题,并且使用队列还可以保证排序算法的稳定性。

但是,这也不是 真正的计数排序,计数排序中使用了一种更巧妙的方法解决这个问题

真正的计数排序

举个例子,班上有
10 名同学:他们的考试成绩分别是:7,8,9,7,6,7,6,8,6,6,他们需要按照成绩从低到高坐到 0~9 共 10 个位置上。
用计数排序完成这一过程需要以下几步:
1.第一步仍然是计数,统计出: 4名同学考了 6 分,3名同学考了 7分,2名同学考了 8 分,1名同学考了9分;
2.然后从头遍历数组: 第一名同学考了7分,共有 4个人比他分数低,所以第一名同学坐在 4 号位置(也就是第5个位置)
3.第二名同学考了 8 分,共有 7 个人 (4 + 3) 比他分数低,所以第二名同学坐在 7 号位置;
4.第三名同学考了 9分,共有 9 个人 (4 + 3 + 2)比他分数低,所以第三名同学坐在 9 号位置;
5.第四名同学考了7分,共有 4 个人比他分数低,并且之前已经有一名考了 7分的同学坐在了 4 号位置,所以第四名同学坐在5号位置。
6…依次完成整个排序

区别就在于计数排序并不是把计数数组的下标直接作为结果输出,而是通过计数的结果,计算出每个元素在排序完成后的位置,然后将元素赋值到对应位置

void countingSort(std::vector<int>& arr) {
	// Check for empty or single-element array
	//判空及防止数组越界
	if (arr.empty() || arr.size() <= 1) return;

	// Find the maximum and minimum values in the array
	//找到最大值,最小值
	int max = arr[0];
	int min = arr[0];
	for (int i = 1; i < arr.size(); i++) {
		if (arr[i] > max) max = arr[i];
		else if (arr[i] < min) min = arr[i];
	}

	// Determine the counting array size
	//确定计数范围
	int range = max - min + 1;

	// Create a counting array of length 'range'
	//建立长度为 range 的数组,下标 0~range-1 对应数字 min~max
	std::vector<int> counting(range, 0);

	// Iterate through the input array and count the occurrences of each element
	//遍历 arr 中的每个元素
	for (int element : arr) {
		// 将每个整数出现的次数统计到计数数组中对应下标的位置,这里需要将每个元素减去 min,才能映射到 0~range-1 范围内
		counting[element - min]++;
	}

	
	// Calculate the cumulative counts
	//记录前面比自己小的数字的总数
	int preCounts = 0;
	for (int i = 0; i < range; i++) {
		// 当前的数字比下一个数字小,累计到 preCounts 中
		preCounts += counting[i];
		// 将 counting 计算成当前数字在结果中的起始下标位置。位置 = 前面比自己小的数字的总数。
		counting[i] = preCounts - counting[i];
	}

	// Create a result array
	std::vector<int> result(arr.size());

	// Place elements in the result array according to their counts
	for (int element : arr) {
		// counting[element - min] 表示此元素在结果数组中的下标
		result[counting[element - min]] = element;
		// 更新 counting[element - min],指向此元素的下一个下标
		counting[element - min]++;
	}

	// Copy the sorted result back to the original array
	//将结果赋值返回arr
	for (int i = 0; i < arr.size(); i++) {
		arr[i] = result[i];
	}
}

倒序遍历的计数排序

void countingSort(std::vector<int>& arr) {
	// Check for empty or single-element array
	//判空及防止数组越界
	if (arr.empty() || arr.size() <= 1) return;

	// Find the maximum and minimum values in the array
	//找到最大值,最小值
	int max = arr[0];
	int min = arr[0];
	for (int i = 1; i < arr.size(); i++) {
		if (arr[i] > max) max = arr[i];
		else if (arr[i] < min) min = arr[i];
	}

	// Determine the counting array size
	//确定计数范围
	int range = max - min + 1;

	// Create a counting array of length 'range'
	//建立长度为 range 的数组,下标 0~range-1 对应数字 min~max
	std::vector<int> counting(range, 0);

	// Iterate through the input array and count the occurrences of each element
	//遍历 arr 中的每个元素
	for (int element : arr) {
		// 将每个整数出现的次数统计到计数数组中对应下标的位置,这里需要将每个元素减去 min,才能映射到 0~range-1 范围内
		counting[element - min]++;
	}


	//每个元素在结果数组中的最后一个下标位置 = 前面比自己小的数字的总数 + 自己的数量 - 1。我们将 counting[0] 先减去 1,后续 counting 直接累加即可

	//记录前面比自己小的数字的总数
	counting[0]--;
	for (int i = 1; i < range; i++) {
		// 将 counting 计算成当前数字在结果中的最后一个下标位置。位置 = 前面比自己小的数字的总数 + 自己的数量 - 1
        // 由于 counting[0] 已经减了 1,所以后续的减 1 可以省略。
		counting[i] +=  counting[i-1];
	}

	// Create a result array
	std::vector<int> result(arr.size());


	// 从后往前遍历数组,通过 counting 中记录的下标位置,将 arr 中的元素放到 result 数组中
	for (int i = arr.size() - 1; i >= 0; i--) {
		// counting[arr[i] - min] 表示此元素在结果数组中的下标
		result[counting[arr[i] - min]] = arr[i];
		// 更新 counting[arr[i] - min],指向此元素的前一个下标
		counting[arr[i] - min]--;
	}



	// Copy the sorted result back to the original array
	//将结果赋值返回arr
	for (int i = 0; i < arr.size(); i++) {
		arr[i] = result[i];
	}
}
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值