寻找最小的k个数(四种方法)

1 使用从大到小的优先队列保存最小的K个数,每次取出K个数之后的其余数和堆顶元素比较,如果比堆顶元素小,则将堆顶元素删除,将该元素插入

void topK(int arr[],int n,int k)
{
    if(k>n)
        return;
    priority_queue<int> q;
    for(int i=0;i<n;++i)
    {
        if(q.size()<k)
            q.push(arr[i]);
        else
        {
            if(arr[i]<q.top())
            {
                q.pop();
                q.push(arr[i]);
            }
        }
    }
    while(!q.empty())
    {
        cout<<q.top()<<‘ ‘;
        q.pop();
    }
    cout<<endl;
}

2 使用set的排序功能,以从大到小的顺序排序所有前K个元素,取出其余的元素与第一个元素比较,如果小于第一个元素,则将第一个元素删除,将当前元素插入。

void topK1(int arr[],int n,int k)
{
    set<int,greater<int> > st;
    if(k>n)
        return;
    for(int i=0;i<n;++i)
    {
        if(st.size()<k)
            st.insert(arr[i]);
        else
        {
            auto iter=st.begin();
            if(*iter>arr[i])
            {
                st.erase(iter);
                st.insert(arr[i]);
            }
        }
    }
    auto iter=st.begin();
    while(iter!=st.end())
    {
        cout<<*iter++<<‘ ‘;
    }
    cout<<endl;
}

3 使用partition的方法,每次找出一个数的固定位置index,其中左边的元素都比该元素小,右边的元素都比该元素大。如果index==k-1,则结束循环,找出的index及之前的元素就是k个最小的元素,否则如果index

大于k-1,则需要找的元素在index的左边,否则需要找的元素在index的右边。循环查找直到index==k-1。

int partition(int arr[],int s,int e)
{
    int priov=arr[e];
    int i=s-1;
    int j=s;
    for(;j<e;++j)
    {
        if(arr[j]<priov)
        {
            ++i;
            swap(arr[i],arr[j]);
        }
    }
    ++i;
    swap(arr[i],arr[e]);
    return i;
}
void topK2(int arr[],int n,int k)
{
    int index=partition(arr,0,n-1);
    while(index!=k-1)
    {
        if(index<k-1)
            index=partition(arr,index+1,n-1);
        else
            index=partition(arr,0,index-1);
    }
    for(int i=0;i<k;++i)
        cout<<arr[i]<<‘ ‘;
    cout<<endl;
}

4 利用选择排序的功能,n个元素中找出最小值与第一个元素交换,从n-1个元素中找出次小值与第二个元素交换,直到找到k个元素位置。

void topK3(int arr[],int n,int k)
{
    for(int i=0;i<k;++i)
    {
        int min=i;
        int j=i+1;
        while(j<n)
        {
            if(arr[j]<arr[min])
                min=j;
            ++j;
        }
        if(min!=i)
            swap(arr[min],arr[i]);
    }
    for(int i=0;i<k;++i)
        cout<<arr[i]<<‘ ‘;
    cout<<endl;
}
时间: 2024-08-29 20:22:44

寻找最小的k个数(四种方法)的相关文章

寻找最小的K个数

寻找最小的K个数 题目描述:查找最小的K个数 题目:输入n个整数,输出其中最小的K个数 例如,输入1.2.3.4.5.6.7.8这8个数字,则最小的4个数字为1.2.3.4. 第一节.各种思路,各种选择 要求一个序列中最小的K个数,按照惯有的思维方式,很简单,先对这个序列从小到大排序,然后输出前面的最小的K个数即可: 至于选取什么样的排序方法,第一时间应该想到的是快速排序,我们知道,快速排序平均时间复杂度为O(nlogn),然后再遍历序列中前K个元素输出,即可,总的时间复杂度为O(nlogn +

寻找最小的k个数(update)

类似编程之美中寻找最大的k个数 解法一: 题目没有要求最小的k个数有序,也没要求最后n-k个数有序.既然如此,就没有必要对所有元素进行排序.这时,咱们想到了用选择或交换排序,即: 1.遍历n个数,把最先遍历到的k个数存入到大小为k的数组中,假设它们即是最小的k个数: 2.对这k个数,利用选择或交换排序找到这k个元素中的最大值kmax(找最大值需要遍历这k个数,时间复杂度为 O(k) ): 3.继续遍历剩余n-k个数.假设每一次遍历到的新的元素的值为x,把x与kmax比较:如果 x < kmax

寻找最小的k个数(大顶堆方法)

题目描述:查找最小的k个元素,输入n个整数,输出其中最小的k个. 一般的排序方法,如快排,时间复杂度为O(n*logn+k); 大顶堆方法,时间复杂度为O(k+(n-k)*logk); 如果建立k个元素的最小堆的话,那么其空间复杂度势为O(N),而建立k个元素的最大堆的空间复杂度为O(k); 当面对海量数据处理的时候,大顶堆的方法是较为靠谱的,并且可以在面试时短时间内完成代码. 1 class Solution { 2 public: 3 void Swap(int &a,int &b)

O(n)时间复杂度求最小的k个数和第k小的数

//思路:使用快速排序的patition函数来进行处理 时间复杂度为O(n) #include<iostream> #include<cassert> using namespace std; int partition(int *ar, int len, int low, int high) { int temp = ar[low]; while(low < high) { while(low < high && temp < ar[high])

C语言K&R习题系列——句子中一个空格代替多个空格的四种方法

原题: Write a program to copy its input to its output, replacing each string of one or more blanks by a single blank. 第一种: 这种最常用,设置一个inspace作为布尔变量,标志当前输入是否在字符中,或在字符外 #include <stdio.h>   int main(void) {   int c;   int inspace=0;     while((c = getcha

寻找最大或最小的K个数

题目描述 :在好几亿个数据中找出最大或最小的K个数. 分析:这几亿的数据肯定不能一起加载到内存中去,更不能对这些数据直接进行排序,因此我们这里讲用数据结构中的 堆 来解决这个问题. 假定这里要从100000个数据中找出最大的100个数据,这样是为了描述方便,我们这里直接用一个数组来存储这个100000个数据,如果数据多达好几亿,则只需将这些数据放入文件中进行读写即可,这里为了描述问题方便就这样假定. 步骤: 取出这些数据中前100个数据,然后用这些数据建立一个小堆: 从第101个数据开始,每读取

剑指offer(二十三,二十四,二十五)最小的k个数,连续子数组的最大和,链表中环的入口节点

23:最小的k个数 题目描述 输入n个整数,找出其中最小的K个数.例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,. 简单题.... function GetLeastNumbers_Solution(input, k) { if(k>input.length) return []; let ans = []; input = input.sort(); //console.log(input.join("").slice(0,4).split

编程之美之2.5 寻找最大的K个数

[题目] 有很多无序的数,从中找出最大的K个数.假定他们都不相等. [解法一] 如果数据不是很多,例如在几千个左右,我们可以排一下序,从中找出最大的K个数.排序可以选择快速排序或者堆排序 [cpp] view plaincopy #include<stdio.h> #include<stdlib.h> int cmp(const void *a,const void *b){ return *(int *)a - *(int *)b; } int main(){ int n,k;

寻找最大的K个数

编程之美有一道考察多种排序的题目,题目如下: 有一个长度为N的无序数组,假定其中的每一个元素都各不相等,求其中最大的K个数. 作者对于此题目结合各种排序算法给出了五种解法思路. 解法一: 使用快速排序或堆排序对它们元素进行排序,整个排序的时间复杂度为O(N*log2N),然后取出前K个,时间复杂度为O(K),总时间复杂度O(N*log2N)+O(K)=O(N*log2N).根据题目要求,并不需要后N-KN-K个数有序,而只需要寻找最大K的个数,因此可以用选择排序和交换排序进行部分排序来筛选最大的