题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
class Solution { public: vector<int> GetLeastNumbers_Solution(vector<int> input, int k) { sort(input.begin(),input.end()); int n=input.size(); vector<int> a; if(n==0 || n==k) return input; if(n > k) { for(int i=0;i<k;i++) a.push_back(input[i]); } return a; } };
另外的思路:利用查找一个数列中第k大的数,只需要建立一个大小为k的最小堆,堆顶就是第k大的数。