leetcode215. 数组中的第K个最大元素(手写最小堆)

题目

给定整数数组 nums 和整数 k，请返回数组中第 k 个最大的元素。

请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

示例 1:

输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:

输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4

题解

本体采用最小堆的思路，限制堆的大小为k，不断地填入数据，替换堆顶，重构堆结构，所有数据处理完后，堆顶就是第k大的结果

class Solution {
    
        private void swap(int[] nums, int index1, int index2){
            int tempt = nums[index1];
            nums[index1] = nums[index2];
            nums[index2] = tempt;
        }

        //初始化堆
        private void buildheap(int[] nums, int len){
            for(int i = len/2 - 1;i >= 0; i--){
                heapify(nums, len, i);
            }
        }

        //建立小顶堆，父节点的下标为i,左右孩子为2i+1, 2i+2
        private void heapify(int[] nums, int len, int i){
            //临时变量minPos用于存储最小值的下标，先假设父节点最小
            int minPos = i;
            while(true){
                //和左节点比较,当左子节点小时，将二者交换位置
                if(2 * i + 1 < len && nums[2 * i + 1] < nums[i]){
                    minPos = 2 * i + 1;
                }
                //和右节点比较,当右子节点小时，将二者交换位置
                if(2 * i + 2 < len && nums[2 * i + 2] < nums[minPos]){
                    minPos = 2 * i + 2;
                }
                if(minPos == i){  //未发生变化则可以直接退出
                    break;
                }
                //否则交换节点
                swap(nums, i, minPos);
                i = minPos;
                //若经过一轮调整后被换下的父节点，依然大于其子节点，则继续调换，直到稳定
            }
        }
    public int findKthLargest(int[] nums, int k) {
        //建立一个堆，包含k个元素
        buildheap(nums, k);
        //遍历余下的元素
        for(int i = k; i < nums.length;i++){
            //比堆顶小的忽略，比堆顶大的加入,并将堆顶挤出
            if(nums[0] > nums[i]){
                continue;
            }
            swap(nums, i, 0);
            heapify(nums, k, 0);

            }
            return nums[0];
        }
    }