您的位置:首页 > 运维架构

TOP-K排序算法,从海量不重复数据中找出最大/小的K个数

2015-04-23 12:08 148 查看
如题,TOP-K排序的主要功能是找出一堆不重复数据中的最小或最大的几个数,此处我们介绍这种类型题目的某种解法:

最大最小堆,最大堆结构里面的每一个数不都是小于root的值么?和我们要解决的问题很像。由此,我们可以构造一个堆,并且用它来存储我们需要找的那几个数。有这么一个动态flash就很好地揭示了其中关系:
top-k动画演示

关于什么是堆在此就不做介绍了,但注意,此处所用的的最大最小堆和数据结构的推排序的堆有区别,不再赘述。下面给出一个例子来供参考:(C++代码)

#include<stdio.h>
#include<iostream>
using namespace std;
struct min_heap{
min_heap(int n){
this->size=n;
for(int i=0;i<size;i++) heap[i]=-1;
}
int size;
int heap[100];
void add(int n){
if(n<=heap[0]) return;
int tem=0;
heap[0]=n;
while(tem<size){
if(2*tem+1>=size) break;
if(2*tem+2>=size&&heap[tem]<heap[2*tem+1]) break;
if(2*tem+2>=size&&heap[tem]>heap[2*tem+1]){
int chan=heap[tem];
heap[tem]=heap[2*tem+1];
heap[2*tem+1]=chan;
break;
}
if(heap[tem]<heap[2*tem+1]&&heap[tem]<heap[2*tem+2]) break;
else{
int min= 2*tem+1;
if(heap[min]>heap[2*tem+2]) min=2*tem+2;
int chan=heap[tem];
heap[tem]=heap[min];
heap[min]=chan;
tem=min;
}
}
}
void print(){
for(int i=0;i<size;i++){
if(heap[i]>0) cout<<heap[i]<<" ";
}
cout<<endl;
}

};

//测试用例:
int main(){
int a[12]={145436,2234,113,4,5,6,7,8,9,23,45,67};
min_heap* test=new min_heap(4);
for(int i=0;i<12;i++){
test->add(a[i]);
test->print();
}
test->print();
return 0;
}
 

结果如下图:




举个例子:

有一个1G大小的一个文件,里面每一行是一个词,词的大小不超过16字节,内存限制大小是1M。返回频数最高的100个词。

大数据的题有这么一个模式:先hash缩小,然后hash统计,然后排序。

此题,内存只有2××20b,数据共2××30b,假设每个词皆2××4b,我们可以推出内存一次可以存2××16个数据,总数据2××26个,我们可以hash%1024分成1024堆(hash等于某值就存储之),每堆大小大约1M,在1024堆里面统计词频,然后用上述K-tops求出结果。

当然,也有可能出现某堆不止1m大小的情况,我们需要具体再细分下去,具体如何我们可以分情况讨论。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐