我有以下实现最大堆的Java代码。它运行正常,
数字列表: 4,1,3,2,16,9,10,14,8,7
MaxHeap结果:
16 14 10 8 7 9 3 2 4 1
我想要的是更改此代码,以便通过给出一个索引来使Max-Heap只是数组的一部分。
例如,如果我给index = 4
它不应该影响4个第一元素:4 1 3 2 但是要用剩余的元素16,9,10,14,8,7制作最大堆
所以最后的结果应该是
4 1 3 2 16 14 10 9 8 7
不应该使用其他数组,只能使用给定的数组:
public class Heapify {
public static int[] Arr = {
4, 1, 3, 2, 16, 9, 10, 14, 8, 7
};
public static int counter = 0;
public static void main(String[] args) {
heapMe();
for (int krk = 0; krk < Arr.length; krk++) {
System.out.println(Arr[krk]);
}
}
public static void heapMe() {
int kk;
for (kk = (Arr.length) / 2 - 1; kk >= 0; kk--) {
heapify(Arr, kk);
}
}
public static void heapify(int[] Arr, int i) {
int largest;
int left = 2 * i + 1;
int right = 2 * i + 2;
if (((left < Arr.length) && (Arr[left] > Arr[i]))) {
largest = left;
} else {
largest = i;
}
if (((right < Arr.length) && (Arr[right] > Arr[largest]))) {
largest = right;
}
if (largest != i) {
swap(i, largest);
heapify(Arr, largest);
}
}
private static void swap(int i, int largest) {
int t = Arr[i];
Arr[i] = Arr[largest];
Arr[largest] = t;
}
}
答案 0 :(得分:2)
您需要为引用n
的函数添加额外参数Arr.length
,并将所有对Arr.length
的引用替换为n
。当您将4
传递给heapMe
时,它会在前四个元素上堆积:
public static void heapMe(int n) {
int kk;
for (kk = n / 2 - 1; kk >= 0; kk--) {
heapify(Arr, n, kk);
}
}
public static void heapify(int[] Arr, int n, int i) {
... // Replace Arr.length with n throughout the body of the function
}