最大堆即大根堆 -- 求前n小 最小堆即小根堆 -- 求前n大 比如求前n小，對于當前元素，和最大堆的堆頂元素（即堆最大值）比較，如果小于堆頂元素，則替換堆頂元素，并調整堆。這樣掃描一遍就可以得到最小的n個元素，效率非常高。 問題實例： 100w個數中找最大的前100個數 1 INT_MIN初始化最小堆 2 讀入一個數，如果大于堆頂元素，則替換堆頂元素，并調整堆 //base index 1
#define LCHILD(i) (2*(i))
#define RCHILD(i) (2*(i)+1)

enum {
BIGHEAP=0,
SMALLHEAP=1,
};

//Adjust the heap
template <typename T>
void HeapAdjust(T *heap, size_t n, bool flg)
{
int i=1;
T root = heap[1];
while (i<n)
{
T ex;
if (flg == BIGHEAP) //最大堆
{
if (LCHILD(i)<=n)
{
ex = LCHILD(i);
if (RCHILD(i)<=n && heap[LCHILD(i)]<heap[RCHILD(i)])
ex = RCHILD(i);
}
else
break;

if (heap[ex] > root)
{
heap[i] = heap[ex];
i = ex;
}
else
break;
}
else //最小堆
{
if (LCHILD(i)<=n)
{
ex = LCHILD(i);
if (RCHILD(i)<=n && heap[LCHILD(i)]>heap[RCHILD(i)])
ex = RCHILD(i);
}
else
break;
if (heap[ex] < root)
{
heap[i] = heap[ex];
i = ex;
}
else
break;
}
}
heap[i] = root;
}

void FindNNumber(const char *file, size_t n, bool flg)
{
int *heap = new int[n+1];
int i;
for (i=1; i<=n; i++)
{
if (flg == BIGHEAP)
heap[i] = INT_MAX;
else
heap[i] = INT_MIN;
}

FILE *fp = fopen(file, "r");

int a;
while(fscanf(fp, "%d", &a)>0)
{
if (flg == BIGHEAP)
{
if (a < heap[1])
heap[1] = a;
}
else
{
if (a > heap[1])
heap[1] = a;

}
HeapAdjust(heap, n, flg);
}
fclose(fp);
for (i=1; i<=n; i++)
printf("%d\n", heap[i]);
}

void main()
{
//最小的n個數
FindNNumber("temp.txt", 10, BIGHEAP);
printf("\n\n");
//最大的n個數
FindNNumber("tt.txt", 10, SMALLHEAP);

}

總結

以上是生活随笔為你收集整理的海量数据前N大的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。