This commit is contained in:
krahets
2023-02-07 17:17:14 +08:00
parent ecbf2d1560
commit 1ca5c731f7
19 changed files with 501 additions and 192 deletions

View File

@@ -260,24 +260,16 @@ comments: true
=== "Java"
```java title="my_heap.java"
// 使用列表而非数组,这样无需考虑扩容问题
List<Integer> maxHeap;
/* 构造函数,建立空堆 */
public MaxHeap() {
maxHeap = new ArrayList<>();
}
/* 获取左子结点索引 */
int left(int i) {
return 2 * i + 1;
}
/* 获取右子结点索引 */
int right(int i) {
return 2 * i + 2;
}
/* 获取父结点索引 */
int parent(int i) {
return (i - 1) / 2; // 向下整除
@@ -408,7 +400,7 @@ comments: true
```java title="my_heap.java"
/* 访问堆顶元素 */
public int peek() {
int peek() {
return maxHeap.get(0);
}
```
@@ -512,13 +504,13 @@ comments: true
// 从底至顶堆化
siftUp(size() - 1);
}
/* 从结点 i 开始,从底至顶堆化 */
void siftUp(int i) {
while (true) {
// 获取结点 i 的父结点
int p = parent(i);
// “越过根结点”或“结点无需修复”,结束堆化
// “越过根结点”或“结点无需修复”,结束堆化
if (p < 0 || maxHeap.get(i) <= maxHeap.get(p))
break;
// 交换两结点
@@ -708,7 +700,7 @@ comments: true
// 返回堆顶元素
return val;
}
/* 从结点 i 开始,从顶至底堆化 */
void siftDown(int i) {
while (true) {
@@ -718,7 +710,7 @@ comments: true
ma = l;
if (r < size() && maxHeap.get(r) > maxHeap.get(ma))
ma = r;
// 若结点 i 最大”或“越过叶结点”,则结束堆化
// 若结点 i 最大或索引 l, r 越界,则无需继续堆化,跳出
if (ma == i) break;
// 交换两结点
swap(i, ma);
@@ -902,7 +894,7 @@ comments: true
```java title="my_heap.java"
/* 构造函数,根据输入列表建堆 */
public MaxHeap(List<Integer> nums) {
MaxHeap(List<Integer> nums) {
// 将列表元素原封不动添加进堆
maxHeap = new ArrayList<>(nums);
// 堆化除叶结点以外的其他所有结点