每日一题1:查找前k个最小值

本文探讨了如何在数据集中有效地找出前k个最小的元素。通过讲解经典的算法和数据结构,如最小堆和二叉查找树,阐述了解决这类问题的关键思路。同时,文章提供了具体的实现细节和实例,适合准备面试和提升算法能力的读者参考。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

 查找前k个最小值最直接的方式是遍历输入数组k遍,每次找出剩下输入中的最小值,每次查找过程中采用交换的策略,这样程序运行结束原数组的前k个数就是按顺序排列的前k个最小数,第二种思路是维护一个具有k个元素的查找树(初始化为输入数组的前k个数),对输入数组的后续每个元素a,将其与查找树的最大数b比较,如果a>=b,则什么也不做,如果a < b,则将b删除,再将a插入到查找树中,如此即可在O(n+nlogk)时间内找到输入数组中的前k个最小值。第三种思路是在原数组上运行k遍初始化最小堆算法,第四种思路,如果输入是整数,则可以使用位图数据结构。下面是第二种思路的实现:
#include "stdafx.h"

#include <set>
#include <iostream>

using namespace std;

bool find_min_k(int input[],int result[],int input_count,int output_count)
{
    if(input_count < output_count) return false;
    set<int> s;
    for (int i = 0; i < output_count; ++i)
    {
        s.insert(input[i]);
    }
    for (int i = output_count; i < input_count; ++i)
    {
        set<int>::iterator tempIt = --s.end();
        if(input[i] < *tempIt)
        {
            s.erase(tempIt);
            s.insert(input[i]);
        }
    }
    int i = 0;
    for (set<int>::iterator it = s.begin(); it != s.end(); ++it)
    {
        result[i++] = *it;
    }
    return true;
}

int _tmain(int argc, _TCHAR* argv[])
{
    int myarray[] = {10,6,15,4,8,1,17,14,5,13,7,11,9,12,16};
    int m = 15;
    int n = 7;
    int *result = new int[n];
    if(find_min_k(myarray,result,m,n))
    {
        for (int i = 0; i < n; ++i)
        {
            cout<<result[i]<<endl;
        }
    }
    return 0;
}

程序运行结果截图

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值