js(63)~215. 数组中的第K个最大元素

昨天,写了几个最大最小极值问题,看解析用的都是堆,看得我一脸迷,我又去看了船长讲的堆的基本介绍,才知道这些题解,都是最小堆,最大堆问题.我今天特意看了别人写的最小堆和最大堆,重点联系,现阶段是能理解的情况下比这瞧一遍代码先

var findKthLargest = function (nums, k) {
	// 从数组中取前 k 个数（ 0 到 k-1 位），构造一个小顶堆, 也就是最大的k个数组成的小顶堆, 堆顶元素就是 第k个最大的数
	// 注意heap 不能初始为空数组
	let heap = [0], i = 0;
	while (i < k) {
		heap.push(nums[i++])
	}
	buildHeap(heap, k);

	// 从 k 位开始遍历数组，每一个数据都和小顶堆的堆顶元素进行比较，如果小于堆顶元素，则不做任何处理，继续遍历下一元素；
	for (let i = k; i < nums.length; i++) {
		const num = nums[i];
		// 如果大于堆顶元素，则将这个元素替换掉堆顶元素，然后再堆化成一个小顶堆。
		if (num > heap[1]) {
			// 替换并堆化
			heap[1] = num;
			heapify(heap, k, 1);
		}
	}

	return heap[1];
};
// 从最后往前,自上而下建造小顶堆
let buildHeap = (arr, k) => {
	if (k === 1) {
		return;
	}
	// 从最后一个非叶子节点开始,自上而下式堆化
	for (let i = Math.floor(k / 2); i >= 1; i--) {
		heapify(arr, k, i)
	}
}

// 堆化
let heapify = (arr, k, i) => {
	while (true) {
		let minIndex = i;
		// minIndex相当于跟节点序号 2*i 相当于左子节点 对比左子节点的序号没有比k数字大 并且跟节点比左子节点大 就根左交换
		if (2 * i <= k && arr[2 * i] < arr[minIndex]) {
			minIndex = 2 * i;
		}
		// 同理对比右和根
		if (2 * i + 1 <= k && arr[2 * i + 1] < arr[minIndex]) {
			minIndex = 2 * i + 1;
		}
		if (minIndex !== i) {
			swap(arr, i, minIndex);
			i = minIndex;
		} else {
			break;
		}

	}
}

let swap = (arr, i, j) => {
	let temp = arr[i];
	arr[i] = arr[j];
	arr[j] = temp;
}