机试之排序——HDU1425sort计数排序

机试之排序——HDU1425sort计数排序

相关链接：
《算法笔记》4.1小节——算法初步->排序

1、题目

题目链接：
http://acm.hdu.edu.cn/showproblem.php?pid=1425

sort
Time Limit: 6000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 82933    Accepted Submission(s): 21450


Problem Description
给你n个整数，请按从大到小的顺序输出其中前m大的数。
 

Input
每组测试数据有两行，第一行有两个数n,m(0<n,m<1000000)，第二行包含n个各不相同，且都处于区间[-500000,500000]的整数。
 

Output
对每组测试数据按从大到小的顺序输出前m大的数。
 

Sample Input
5 3
3 -35 92 213 -644
 

Sample Output
213 92 3

Hint
请用VC/VC++提交

2、代码

本题主要是在时间限制上，都不可以用流输入输出cin、cout，必须采用C的格式化输入，既可以直接用sort（）函数也可以用计数排序

2.1直接sort()函数排序

//计数排序
//HDU1425sort 
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAXN = 1e6 + 10;
//const int MAXN = 1000000;
int arr[MAXN];		//数据数组 

int main(){
	int n,m;
	while(cin>>n>>m){ 
		for(int i = 0; i < n; ++i){
			scanf("%d",&arr[i]);
	//		cin>>arr[i];
		}
		sort(arr,arr+n);
		for(int i = n-1; i >= n - m; --i){
			if(i == n-m){
				printf("%d\n",arr[i]);
	//			cout<<arr[i]<<endl;
			}
			else{
				printf("%d ",arr[i]);
		//		cout<<arr[i]<<" ";
			}
		}
	} 
    return 0;
}

2.2计数排序

//计数排序法
//HDU1425sort 
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAXN = 1e6 + 10;
const int shift = 5e5;//将数据区间[-500000,500000]平移到 [0,1000000]，因为number计数数组下标从0开始 
//const int MAXN = 1000000;
int arr[MAXN];		//数据数组 
int number[MAXN];	//计数数组 

int main(){
	int n,m;
	while(cin>>n>>m){
		memset(number,0,sizeof(number));//初始化计数数组 
		for(int i = 0; i < n; ++i){
			scanf("%d",&arr[i]);
	//		cin>>arr[i];
			number[arr[i] + shift ]++;
		}
		int index = 0;
		for(int i = 0; i < MAXN; ++i){
			while(number[i]--){
				arr[index++] = i - shift;
			}
		}

		for(int i = n-1; i >= n - m; --i){
			if(i == n-m){
				printf("%d\n",arr[i]);
	//			cout<<arr[i]<<endl;
			}
			else{
				printf("%d ",arr[i]);
		//		cout<<arr[i]<<" ";
			}
		}
	} 
    return 0;
}