因为堆结构只保证 根节点比双子节点都大或小

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. 因为资源和程序源码均为可复制品,所以不支持任何理由的退款兑现,请斟酌后支付下载 声明:如果标题没有注明"已测试"或者"测试可用"等字样的资源源码均未经过站长测试.特别注意没有标注的源码不保证任何可用性