因为堆结构只保证 根节点比双子节点都大或小
1 求最小的n个数:
构建n个数的大顶堆,依次弹出堆顶再往下调整(用例省略)
2 求最大的n个数:
构建n个数的小顶堆,依次弹出堆顶再往下调整(用例省略)
3 求第n大的数:
构建所有元素的大顶堆,依次弹出堆顶,堆底放到堆顶并向下调整,重复n次
用例:
public int findKth(int[] a, int n, int K) { ArrayList<Integer> list = (ArrayList<Integer>)Arrays.stream(a).boxed().collect( Collectors.toList()); //大根堆 buildHeap(list); int maxK = 0; for (int i = 0 ; i < K ; i++) { maxK = list.get(0); //顶换到底部 swap(list, list.size() - 1, 0); //删除队尾元素 list.remove(list.size() - 1); //向下调整 shiftDown(list, 0); } return maxK; } private void buildHeap(ArrayList<Integer> a) { int len = a.size(); for (int i = len / 2 - 1 ; i >= 0; i--) { shiftDown(a, i); } } private void shiftDown(ArrayList<Integer> a, int i) { System.out.println(a.size() + ";" + i); int len = a.size() - 1; int maxIndex = 0; if (2 * i + 2 > len) { //没有右子节点 if(2*i+1 > len) { return; }else{ //有左子节点 maxIndex = 2 * i + 1; } } else { //有双子节点 if (a.get(2 * i + 1) < a.get(2 * i + 2)) { //左子节点小于右子节点 maxIndex = 2 * i + 2; } else { //左子节点大 maxIndex = 2 * i + 1; } } if (a.get(i) < a.get(maxIndex)) { swap(a, i, maxIndex); shiftDown(a, maxIndex); } } private void swap(ArrayList<Integer> a, int i, int j) { int temp = a.get(i); a.set(i, a.get(j)); a.set(j, temp); }
原文地址:http://www.cnblogs.com/yanher/p/16885315.html
1. 本站所有资源来源于用户上传和网络,如有侵权请邮件联系站长!
2. 分享目的仅供大家学习和交流,请务用于商业用途!
3. 如果你也有好源码或者教程,可以到用户中心发布,分享有积分奖励和额外收入!
4. 本站提供的源码、模板、插件等等其他资源,都不包含技术服务请大家谅解!
5. 如有链接无法下载、失效或广告,请联系管理员处理!
6. 本站资源售价只是赞助,收取费用仅维持本站的日常运营所需!
7. 如遇到加密压缩包,默认解压密码为"gltf",如遇到无法解压的请联系管理员!
8. 因为资源和程序源码均为可复制品,所以不支持任何理由的退款兑现,请斟酌后支付下载
声明:如果标题没有注明"已测试"或者"测试可用"等字样的资源源码均未经过站长测试.特别注意没有标注的源码不保证任何可用性