最小的K个数 (冒泡和最小堆）

最新推荐文章于 2021-05-18 21:44:31 发布

「已注销」

最新推荐文章于 2021-05-18 21:44:31 发布

阅读量620

点赞数

分类专栏： ACM 文章标签：冒泡排序最小堆

本文链接：https://blog.csdn.net/hyqsong/article/details/49530115

版权

ACM 专栏收录该内容

29 篇文章 1 订阅

订阅专栏

问题：最小的K个数

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。

方法一（冒泡排序k趟）

思路：冒泡排序进行k趟即可，就把最小的k个跳出来了

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> res;
        int len = input.size();
        if(len == k )
            return input;
        if(k>len || k<=0) //注意判断条件
            return res;

        //冒泡前k趟即可
        for(int i = 0;i<k;i++)
            {
            for(int j = len-1;j>i;j--)
                {
                if(input[j] < input[j-1])
                    {
                    int temp = input[j];
                    input[j] = input[j-1];
                    input[j-1] = temp;
                }
            }
            res.push_back(input[i]); //放在这儿就行拉，何必再做一遍          
        }

        //思路是这么走的，但是写完后还是要看看怎么简化代码的
        //for(int i =0;i<k;i++)
            //{
          //  res.push_back(input[i]);
        //}
        return res;
    }
};

方法二：堆排序

思路：使用简单粗暴的全部排序的方法并不能取得很好的性能只要求k个数，所以使用最小堆就可以了。(如果是要取最大值就是用最大堆）

class Solution {
private:
     void heapSort(vector<int> &input, int root, int end){
        for(int j = end -1; j >= root; j --){
            int parent = (j + root -1)/2;
            if(input[parent] > input[j]){
                int temp = input[j];
                input[j] = input[parent];
                input[parent] = temp;
            }
        }   
     }

public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> result ;
        if(k > input.size()) return result;
        for(int i = 0; i < k ; i ++){
            heapSort(input,i,input.size());
            result.push_back(input[i]);
        }
        return result;
    }
};