剑指offer二十九---最小的k个数

Posted 2021-03-17 linxuesong

Markdown在线编辑器 - www.MdEditor.com

1.方法一：借助辅助数组存储k个最小的数

思想

存着最小k个数的数组，内部有序，遍历所有元素，和辅助数组中最大的比，只要小就替换辅助数组中的最大元素，然后再排序

代码

1. // 使用辅助数组来实现
2. vector<int> FuZhu(vector<int> a,int k) {
4. vector<int> result;
5. if(a.size() < k || k == 0) return result;
6. if (a.empty()) {
7. return result;
8. }
9. result.push_back(a[0]);
10. for (int i = 1; i < a.size(); i++) {
11. if (result.size() == k) {
12. sort(result.begin(), result.end());
13. if (a[i] < *(result.end()-1)) {
14. result.pop_back();
15. result.push_back(a[i]);
16. }
17. } else {
18. result.push_back(a[i]);
19. }
20. }
21. //时间复杂度
22. return result;
23. }
25. vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
26. return FuZhu(input, k);
27. }

2.方法二：借助快速排序找到k的位置

思想

快速排序可以保证k左边的都小，右边都大，当基准元素位置==k个数时，输出前k个即可。
不用排两边，只要照着k在的那一侧即可

代码

1. intOnce(vector<int>&a,int start,intend){
2. int mark = start;
3. int tmp =0;
4. for(int i = start +1; i <=end; i++){
5. if(a[start]> a[i]){
6. mark++;
7. tmp = a[i];
8. a[i]= a[mark];
9. a[mark]= tmp;
10. }
11. }
12. tmp = a[mark];
13. a[mark]= a[start];
14. a[start]= tmp;
15. return mark;
16. }
17. void kuaisu(vector<int>&a,int k,int start,intend){
18. int i =Once(a, start,end);
19. if(k == i){
20. return;
21. }
22. // 这次分完位置在i，和k去比较
23. if(i > k){
24. kuaisu(a, k,0, i -1);
25. }
26. else{
27. kuaisu(a, k, i +1,end);
28. }
29. }
31. // 借鉴快速排序实现
32. vector<int>GetLeastNumbers_Solution(vector<int> input,int k){
34. //找到k的位置就行
35. vector<int> result;
36. if(k > input.size())return result;// 找不到k位置
37. if(input.empty()){
38. return result;
39. }
40. // 让我们开始快速排序
41. kuaisu(input, k,0, input.size()-1);
42. // 把前k个数存入result
43. for(int i =0; i < k; i++){
44. result.push_back(input[i]);
45. }
46. return result;
47. }