30-最小的K个数

最新推荐文章于 2024-07-15 07:43:36 发布

leechow007

最新推荐文章于 2024-07-15 07:43:36 发布

阅读量224

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/leechow007/article/details/74559375

本文介绍了一种利用快速排序的partition过程和最大堆数据结构来高效找到数组中最小的K个数的方法。通过具体算法实现，展示了如何在不完全排序的情况下找到所需元素，适用于大数据集的Top K问题。

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

一利用快排 partition

轴值的下标是k-1的时候，说明前面的K个数就是所求。

vector<int> GetLeastNumbers_Solution(vector<int> input, int k)
    {
        vector<int> result;
        
       if (0 >= input.size() || k > input.size()||k<1)
           return result;
        int l=0;
        int r=input.size()-1;
        int index= partition(input,l,r);
        while(index!=k-1)
            if(index<k-1)
            {
                l=index+1;
                index= partition(input,l,r);
            }
            else
            {
                r=index-1;
                index= partition(input,l,r);
            }
         vector<int> res(input.begin(), input.begin() + k);
         
        return res;
       
        
    }
    int partition(vector<int> &s,int l,int r)
        {
        int i=l;
        int j=r;
        int x=s[i];
        while (i<j)
        {
            while(i<j&&s[j]>=x)
                j--;
            if(i<j)
                {
                   s[i]=s[j];
                   i++;
                }
             while(i<j&&s[i]<x)
                i++;
                if(i<j)
                {
                   s[j]=s[i];
                   j--;
  
               }
        }
        s[i]=x;
        return i;
    }