topK问题-在海量数据从大到小输出前topk个数据

最新推荐文章于 2022-02-09 17:04:37 发布

Ocodotial

最新推荐文章于 2022-02-09 17:04:37 发布

阅读量156

点赞数

分类专栏：本科的学习文章标签：算法 c++ 数据结构

本文链接：https://blog.csdn.net/qq_43656353/article/details/108936864

版权

本科的学习专栏收录该内容

37 篇文章 14 订阅

订阅专栏

本文介绍了一个C++程序，用于处理大量正整数输入，实时计算并输出按逆序排列的前k个数。通过堆数据结构实现动态更新和快速查找，适用于实时数据流处理场景。用户可以连续输入数据，程序会即时返回最新top K结果。

摘要由CSDN通过智能技术生成

求海量数据(正整数)按照逆序排列的前k个数(topK)。
不限制用户输入数据个数，用户每输入一个数据就回车使得程序可立即获得这个数据，用户输入-1代表输入终止

#include <cstdio>
#include <iostream>
using namespace std;
#define N 1000
/**
 * 前k个数
 */
int index_e = 0, k, heap[N], x;
void heapJust(int arr[], int s, int t);
void heapMin(int arr[], int s, int t);
void deal(int x);
void printRs();

/**
 * 如果数据数量小于等于k，直接加入堆中
 * 等于k的时候，进行堆化
 **/
void deal(int x)
{
    if(index_e < k-1)
    {
        heap[index_e++] = x;
    }
    else if(index_e == k-1)   //对heap数组进行堆化
    {
        heap[index_e++] = x;
        heapMin(heap, 0, k-1);
//        printRs();
    }
    //x和堆顶进行对比,如果x大于堆顶，x将堆顶挤掉并且向下调整
    else if(index_e > k-1 && heap[k-1] < x)
    {
        heap[k-1] = x;
        heapMin(heap, 0, k-1);
        printRs();
    }
}

void printRs()
{
    for(int i = 0; i < k; i++)
    {
        printf("%d ", heap[i]);
    }
    printf("\n");
}
void heapJust(int arr[], int s, int t)
{
    int temp = arr[s];
    for(int i = 2*s; i <= t; i*=2)
    {
        if(i < t && arr[i] > arr[i+1])
        {
            i++;
        }
        if(arr[i] >= temp)
        {
            break;
        }
        arr[s] = arr[i];
        s = i;
    }
    arr[s] = temp;
}
void heapMin(int arr[], int s, int t)
{
    for(int i = t/2; i >= s; i--)
    {
        heapJust(arr, i, t);
    }
    for(int i = t; i > s; i--)
    {
        swap(arr[i], arr[s]);
        heapJust(arr, s, i-1);
    }
}
int main()
{
    scanf("%d", &k);
    scanf("%d", &x);
    while(x != -1)
    {
        deal(x);
        scanf("%d", &x);
    }
    return 0;
}