华为OJ（输出数组最小k个数）

方法1：利用algorithm中sort算法（不造sort的实现是不是就是快速排序）

引自：http://www.geekboy.org/huaweioj-three/

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

bool comp1(const int &a,const int&b)
{
	return a<b;
}
int main()
{
	int N,k;
	cin>>N;
	cin>>k;
	vector<int> array;
	for(int i=0; i<N; i++)
	{
		int m;
		cin>>m;
		array.push_back(m);
	}
	sort(array.begin(),array.end(),comp1);
	for(int i=0; i<k; i++)
	{
		cout<<array[i];
		if(i==k-1)
		{
			cout<<endl;
			continue;
		}
		cout<<' ';
	}
}

方法2：快速排序，输出前k个数，参考剑指offer面试题30

不知道问题出在哪，在OJ系统提交木有通过，提示答案错误，在此把代码贴出来。

#include<iostream>
#include<time.h>
using namespace std;
#define N 100
int RandomInt(int,int);
int Partion(int*,int,int,int);
void swap(int&,int&);
bool GetminK(int,int*,int,int*);
int main()
{
	int pInput[N],poutput[N],inputNum,k;
	cin>>inputNum;
	cin>>k;
	for(int i=0;i<inputNum;++i)
		cin>>*(pInput+i);	
	GetminK(inputNum,pInput,k,poutput);
	//system("pause");
	return 0;
}
void swap(int& a,int &b)
{
	int tmp;
	tmp=a;
	a=b;
	b=tmp;
}
int RandomInt(int m,int n)
{
	srand((unsigned int)time(NULL));
	return (rand()%(n-m))+m;
}
int Partion(int *data,int length,int start,int end)
{
	if(data==NULL||length<=0||start<0||end<start)
		return NULL;
	int index=RandomInt(start,end);
	swap(data[index],data[end]);
	int small=start-1;
	for(int i=start;i<end;++i)
	{
		if(data[i]<data[end])
		{
			small++;
			if(i!=small)
				swap(data[i],data[small]);
		}
	}
	small++;
	swap(data[small],data[end]);
	return small;
}
bool GetminK(int inputNum,int * pInput,int k,int * poutput)
{
	if(inputNum<=0||k<=0)
		return false;
	int start=0,end=inputNum-1;
	int index=Partion(pInput,inputNum,start,end);
	while(index!=k-1)
	{
		if(index<k-1)
			index=Partion(pInput,inputNum,(index+1),end);
		else
			index=Partion(pInput,inputNum,start,(index-1));
	}
	for(int i=0;i<k;++i)
		cout<<*(pInput+i);
	return true;
}

方法3：

参考剑指offer上，基于红黑树，其实就是利用STL中set排序，但还是木有通过OJ测试，不造什么原因。

#include<iostream>
#include<set>
#include<vector>
#include<functional>
using namespace std;
bool GetleastkNum(const vector<int>&,multiset<int,greater<int>>&,int);
int main()
{
	int n;
	int k;
	cin>>n;
	cin>>k;
	vector<int> ivec(n);
	multiset<int,greater<int>> iset;
	for(int i=0;i<n;++i)
		cin>>ivec[i];
	GetleastkNum(ivec,iset,k);
	for(multiset<int,greater<int>>::const_iterator it=iset.end();it!=iset.begin();)
		cout<<*(--it)<<" ";
	cout<<endl;
	//system("pause");
	return 0;
}
bool GetleastkNum(const vector<int>& ivec,multiset<int,greater<int>>& iset,int k)
{
	if(ivec.size()<k||k<=0)
		return false;
	iset.clear();
	multiset<int,greater<int>>::iterator iterlarge;
	for(vector<int>::const_iterator it=ivec.begin();it!=ivec.end();++it)
	{

		if(iset.size()<k)
			iset.insert(*it);
		else
		{
			iterlarge=iset.begin();
			if(*it<*iterlarge)
				{
					iset.erase(iterlarge);
					iset.insert(*it);
				}
		}
	}
	return true;
}