1. 程式人生 > >海量資料處理(一) 求top k問題

海量資料處理(一) 求top k問題

優先順序佇列
給一組海量資料,限制記憶體為2M,,找出裡面最大/小的Tokp k

 


int main()
{
	vector<int> vec;
	srand(time(NULL));
	for(int i =0;i<1000000;i++)
	{
		vec.push_back(rand()%1000000+1);
	}
	//預設是一個大根對
	priority_queue<int, vector<int>, greater<int> > queue;  //小根堆
	srand(time(NULL));
	for(int i =0;i<10;i++)
	{
	queue.push(vec[i]);
	}
	for(int i=0;i<vec.size();++i)
	{
		if(queue.top() < vec[i])
		{
			queue.pop();
			queue.push(vec[i]);
		}
	}
	while(!queue.empty())
	{
	cout<<queue.top()<<endl;
  	queue.pop();
	}
	system("pause");
	return 0;
}