面试经典算法题 LeetCode 215. 数组中的第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

原题链接：https://leetcode.cn/problems/kth-largest-element-in-an-array

---

首先我们需要定义好大根堆的结构，也就是根节点大于左右节点。设某个根节点的索引为 idx 则这个节点的左右节点分别为2 * idx + 1、2 * idx + 2。若根节点小于子节点，则需要与其中最大的子节点交换值。每次调整过后，该子节点的值发生变化，需要继续递归，对子树的结构进行调整。

	void MaxHeap(int[] nums, int n, int idx){
        //调整堆结构
        int left = idx * 2 + 1;//左节点
        int right = idx * 2 + 2;//右节点
        int max = idx;//初始化最大值索引为idx

        if(left < n && nums[max] < nums[left]) max = left;//左节点大于max，max指向左节点位置
        if(right < n && nums[max] < nums[right]) max = right;//右节点大于max，max指向右节点位置
        
        if(max != idx){
        	//堆结构发生变化，父节点与子节点交换
            int temp = nums[idx];
            nums[idx] = nums[max];
            nums[max] = temp;

            MaxHeap(nums, n, max);//调整子树
        }
    }

然后是建堆，为了减少操作，建堆时我们需要从第一个非叶子节点开始从下往上建堆。根据堆结构的性质可知，第一个非叶子节点即为数组总数 n / 2 - 1。

	void BuildMaxHeap(int[] nums, int n){
    	//建堆
        int bot = n / 2 - 1;
        for(int i = bot;i >= 0;i--){
            MaxHeap(nums, n, i);
        }
    }

最后是获取第K大的值，我们只需要每次将堆顶元素与末尾元素交换，然后将该最大值移出堆，不再需要对该值进行堆排序，直到第K大的数成为堆顶时，结束循环。

	public int FindKthLargest(int[] nums, int k) {
        int n = nums.Length;
        BuildMaxHeap(nums, n);

        for(int i = n - 1;i >= n - k + 1;i--){
        	//把当前的根节点（最大值）丢到数组未排序部分最后一项
            int temp = nums[i];
            nums[i] = nums[0];
            nums[0] = temp;

            MaxHeap(nums, i, 0);//调整当前堆结构
        }

        return nums[0];
    }

完整代码如下

public class Solution {
    void MaxHeap(int[] nums, int n, int idx){
        //调整堆结构
        int left = idx * 2 + 1;
        int right = idx * 2 + 2;
        int max = idx;

        if(left < n && nums[max] < nums[left]) max = left;
        if(right < n && nums[max] < nums[right]) max = right;
        
        if(max != idx){
            int temp = nums[idx];
            nums[idx] = nums[max];
            nums[max] = temp;

            MaxHeap(nums, n, max);//调整子树
        }
    }

    void BuildMaxHeap(int[] nums, int n){
    	//建堆
        int bot = n / 2 - 1;
        for(int i = bot;i >= 0;i--){
            MaxHeap(nums, n, i);
        }
    }

    public int FindKthLargest(int[] nums, int k) {
        int n = nums.Length;
        BuildMaxHeap(nums, n);

        for(int i = n - 1;i >= n - k + 1;i--){
        	//把当前的根节点（最大值）丢到数组未排序部分最后一项
            int temp = nums[i];
            nums[i] = nums[0];
            nums[0] = temp;

            MaxHeap(nums, i, 0);//调整当前堆结构
        }

        return nums[0];
    }
}