架构算法设计模式和编程理论程序员数据结构和算法分析

堆排序

2018-06-29  本文已影响2人  宛丘之上兮

要理解堆排序,首先要了解这个数据结构,因为堆排序是利用这种数据结构而设计的一种排序算法。首先是一种完全二叉树(满二叉树、完全二叉树、平衡二叉树、二叉排序树、红黑树or平衡二叉搜索树、哈弗曼树or最优二叉树),其次满足下面的性质:

堆排序

堆排序是一种选择排序,最坏,最好,平均时间复杂度均近似为O(nlogn),是不稳定排序。

算法流程主要包括2部分:1构建初始堆和2交换堆顶元素和末尾元素并重建堆。其中构建初始堆经推导复杂度为O(n),在交换并重建堆的过程中,需交换n-1次,而重建堆的过程中,根据完全二叉树的性质,[log2(n-1),log2(n-2)...1]逐步递减,近似为nlogn。所以堆排序时间复杂度一般认为就是O(nlogn)级。

下面贴上javascript代码:

function sort(arr) {
    //1.构建大顶堆
    for (var i = Math.floor(arr.length / 2) - 1; i >= 0; i--) {
        //从第一个非叶子结点从下至上、从右至左调整结构
        adjustHeap(arr, i, arr.length);
    }
    //2.调整堆结构 && 交换堆顶元素与末尾元素
    for (var j = arr.length - 1; j > 0; j--) {
        swap(arr, 0, j);//将堆顶元素与末尾元素进行交换
        adjustHeap(arr, 0, j);//从第一个结点从下至上、从右至左调整结构
    }
}

function adjustHeap(arr, i, length) {
    var temp = arr[i];
    for (var k = i * 2 + 1; k < length; k = k * 2 + 1) {//从i结点的左子结点开始,也就是2i+1处开始
        if (k + 1 < length && arr[k] < arr[k + 1]) {//如果左子结点小于右子结点,k指向右子结点
            k++;
        }
        if (arr[k] > temp) {//如果子节点大于父节点,将子节点值赋给父节点(不用进行交换)
            arr[i] = arr[k];
            i = k;
        } else {
            break;
        }
    }
    arr[i] = temp;//将temp值放到最终的位置
}

function swap(arr, a, b) {
    var temp = arr[a];
    arr[a] = arr[b];
    arr[b] = temp;
}

var arr = [22, 1, 324, 4, 6, 22, 11, 9, 8, 7, 6, 5, 4, 3, 2, 1];
sort(arr);
console.log(arr)//[ 1, 1, 2, 3, 4, 4, 5, 6, 6, 7, 8, 9, 11, 22, 22, 324 ]

小顶堆排序的话只需要改adjustHead函数中的两行代码:

  1. if (k + 1 < length && arr[k] < arr[k + 1])改为if (k + 1 < length && arr[k] > arr[k + 1])
  2. if (arr[k] > temp)改为if (arr[k] < temp)





参考文献:

  1. 图解排序算法(三)之堆排序
上一篇 下一篇

猜你喜欢

热点阅读