使用紅/黑樹狀結構尋找最小n個數，適合處理大資料

標籤：紅/黑樹狀結構 尋找 最小n個數   適合處理大資料   

使用STL容器 multiset

代碼：

#include <iostream>#include <vector>#include <set>using namespace std;typedef multiset<int,greater<int> > intSet;typedef multiset<int,greater<int> >::iterator intIter;void findMaxNum(const vector<int> &data,int k){intSet Max; Max.clear();if(k < 1 || data.size() < k)    return ;vector<int>::const_iterator iter = data.begin();for(;iter != data.end();++iter){if(Max.size() < k){Max.insert(*iter);}else{intIter least=Max.begin();if((*iter) < *( Max.begin())){ //只能是begin（）刪除Max.erase(least);Max.insert(*iter);}}}for( intIter it = Max.begin(); it != Max.end();it ++){cout<<*it << "  ";}}int main(){vector<int> data;for(int i=0;i<1000;i++){data.push_back(i);}findMaxNum(data,10);    return 0;}

運行結果：