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

来源:互联网 发布:cpu温度监控软件 编辑:程序博客网 时间:2024/06/05 08:16

问题:最小的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;    }};

欢迎补充更好的方法
未完待续。。。。。。。。。。。

0 0