subArray中的Max Heap

时间:2013-02-27 02:55:44

标签: java algorithm heap max heapsort

我有以下实现最大堆的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;
    }
}

1 个答案:

答案 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
}