堆排序

2017-11-29  本文已影响11人  Lemon_Home

堆的定义:一组能够用堆有序的完全二叉树排序的元素,并在数组中按照层级存储(不适用数组第一个位置)
堆有序:当一颗二叉树的每个节点都大于等于它的两个子节点时,它被称为堆有序。

堆排序是一种选择排序,其时间复杂度为O(nlogn)。
实现的大概思路:

堆排序

堆排序是唯一能够同时最优地利用空间和时间的方法——在最坏的情况下它也能保证使用~ 2NlgN次比较和恒定的额外空间。但现代操作系统的许多应用很少使用它,因为它无法利用缓存。

public class Heap {

    public static void sort(Comparable[] a) {
        int N = a.length - 1;
        for (int k = N / 2; k >= 1; k--) {
            sink(a, k, N);
        }
        while (N > 1) {
            exch(a, 1, N--);
            sink(a, 1, N);
        }
    }

    public static void sink(Comparable[] a, int k, int n) {
        while (2 * k <= n) {
            int j = 2 * k;
            if (j < n && less(a[j], a[j + 1])) j++;
            if (!less(a[k], a[j])) break;
            exch(a, k, j);
            k = j;
        }
    }

    public static boolean less(Comparable v, Comparable w) {
        return v.compareTo(w) < 0;
    }

    public static void exch(Comparable[] a, int i, int j) {
        Comparable t = a[i];
        a[i] = a[j];
        a[j] = t;
    }

    public static void show(Comparable[] a) {
        for (int i = 0; i < a.length; i++) {
            System.out.print(a[i] + " ");
        }
        System.out.println();
    }

    public static void main(String[] args) {
        String[] a = {" ", "80", "30", "60", "40", "20", "10", "50", "70"};
        show(a);
        sort(a);
        show(a);
    }
}
class Heap:
    def heap_sort(self, data):
        data.insert(0, "")
        n = len(data) - 1
        k = n // 2
        while k >= 1:
            self.sink(data, k, n)
            k -= 1

        while n > 1:
            self.exch(data, 1, n)
            n -= 1
            self.sink(data, 1, n)
        data.pop(0)

    def sink(self, data, k, length):
        while 2 * k <= length:
            j = k * 2
            if j < length and self.less(data[j], data[j+1]):
                j += 1
            if not self.less(data[k], data[j]):
                break
            self.exch(data, k, j)
            k = j

    def less(self, x, y):
        return x < y

    def exch(self, data, i, min_index):
        data[i], data[min_index] = data[min_index], data[i]


if __name__ == "__main__":
    s = [3, 4, 1, 6, 2, 9, 7, 0, 8, 5]
    heap = Heap()
    heap.heap_sort(s)
    print(s)
上一篇 下一篇

猜你喜欢

热点阅读