当前位置:   article > 正文

7月18日学习打卡,数据结构堆

7月18日学习打卡,数据结构堆

hello大家好呀,本博客目的在于记录暑假学习打卡,后续会整理成一个专栏,主要打算在暑假学习完数据结构,因此会发一些相关的数据结构实现的博客和一些刷的题,个人学习使用,也希望大家多多支持,有不足之处也请指出。本篇只讲解OJ题,关于知识点的博客稍后发出,希望大家多多支持。

面试题17.14,最小k个数

. - 力扣(LeetCode)

思路一:构建小堆

我们知道已经学习过优先级队列PriorityQueue,并且知道它底层就是堆,并且默认是小堆存放,那么,第一种思路便很清晰了,用数据集合中前K个元素来建堆,然后获取堆中前k个元素

  1. class Solution {
  2. public int[] smallestK(int[] arr, int k) {
  3. int[] ret=new int[k];
  4. PriorityQueue<Integer> priorityQueue=new PriorityQueue<>();
  5. for (int i = 0; i <arr.length ; i++) {
  6. priorityQueue.offer(arr[i]);
  7. }
  8. for (int i = 0; i < k; i++) {
  9. ret[i]=priorityQueue.poll();
  10. }
  11. return ret;
  12. }
  13. }

思路二:构建大堆

可以通过构建小堆解题,那么同样也可以使用大堆,第二个思路便是自己实现比较方法构建大堆,将剩余的N-K个元素依次与堆顶元素来比较,如果小于栈顶元素则替换堆顶元素,然后将剩余N-K个元素依次与堆顶元素比完之后,堆中剩余的K个元素就是所求的前K个最小。

  1. class Solution {
  2. public int[] smallestK(int[] arr, int k) {
  3. if(k<=0){
  4. return new int[0];
  5. }
  6. int[] ret=new int[k];
  7. PriorityQueue<Integer> priorityQueue=new PriorityQueue<>(new com());
  8. for (int i = 0; i <k; i++) {
  9. priorityQueue.offer(arr[i]);
  10. }
  11. for (int j = k; j <arr.length; j++) {
  12. if(arr[j]<priorityQueue.peek()){
  13. priorityQueue.poll();
  14. priorityQueue.offer(arr[j]);
  15. }
  16. }
  17. for (int i = 0; i < k; i++) {
  18. ret[i]=priorityQueue.poll();
  19. }
  20. return ret;
  21. }
  22. }
  23. class com implements Comparator<Integer>{
  24. @Override
  25. public int compare(Integer t2, Integer t1) {
  26. return t1-t2;
  27. }
  28. }

以上就是本期博客全部内容啦,明天考完科二会会把PriorityQueue详解发出来,感谢大家支持

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/代码探险家/article/detail/872348
推荐阅读
相关标签
  

闽ICP备14008679号