topk的两种算法C++实现（堆，快速查找）

#include
using namespace std;

int topk(vector arr,int k)//堆方式 
{
	if(k>arr.size()||k<=0)
	{
		return -1;
	}
	priority_queue,greater > qp;
	for(int i=0;i
		if(qp.size()
			qp.push(arr[i]);
		}
		else
		{
			if(qp.top()
				qp.pop();
				qp.push(arr[i]);
			}
		}
	}
	return qp.top();
}

int topk2main(vector& arr,int k,int l,int r)
{
	if(l>r)
	{
		return -1;
	}
	int temp = arr[l];
	int i=l;
	int j=r;
	while(i
		while(i=temp) j--;
		while(i
			swap(arr[i],arr[j]);
		}
	}
	arr[l]=arr[i];
	arr[i]=temp;
	
	if(r-i+1==k)
	{
		return arr[i];
	}
	else if(r-i
		return topk2main(arr,k+i-r-1,l,i-1);
	}
	else
	{
		return topk2main(arr,k,i+1,r);
	}
}

int topk2(vector arr,int k)//快速选择方式 
{
	if(k>arr.size()||k<=0)
	{
		return -1;
	}
	return topk2main(arr,k,0,arr.size()-1);
}

void test(vector arr)
{
	bool flag = false;
	for(int i=0;i<=1000;i++)
	{
		int a=topk(arr,i);
		int b=topk2(arr,i);
		int c=a==b?1:0;
		cout<
			flag=true;
		}
	}
	if(flag)
	{
		cout<<"ERROR!"<
		cout<<"YES!"<
	srand(time(NULL));
	vector arr;
	for(int i=0;i<1000;i++)
	{
		arr.push_back(rand()%2000);
	}
	test(arr);
}