堆排序实现

来源:互联网 发布:网络侵权管辖 编辑:程序博客网 时间:2024/06/18 11:11
--[[(1)用大根堆排序的基本思想① 先将初始文件R[1..n]建成一个大根堆,此堆为初始的无序区② 再将关键字最大的记录R[1](即堆顶)和无序区的最后一个记录R[n]交换,由此得到新的无序区R[1..n-1]和有序区R[n],且满足R[1..n-1].keys≤R[n].key③由于交换后新的根R[1]可能违反堆性质,故应将当前无序区R[1..n-1]调整为堆。然后再次将R[1..n-1]中关键字最大的记录R[1]和该区间的最后一个记录R[n-1]交换,由此得到新的无序区R[1..n-2]和有序区R[n-1..n],且仍满足关系R[1..n-2].keys≤R[n-1..n].keys,同样要将R[1..n-2]调整为堆。直到无序区只有一个元素为止--核心是调整为大根堆结构]]function swap(i1,i2,arr)    local tmp = arr[i2]    arr[i2] = arr[i1]    arr[i1] = tmpend --array是待调整的堆数组,i是待调整的数组元素的位置,nlength是数组的长度--本函数功能是:根据数组array构建大根堆function HeapAdjust(array,i,nLength)    local nChild    --树深度    while( 2*i <= nLength ) do        --子结点的位置=2*(父结点位置)        nChild = 2*i        --得到子结点中较大的结点        if(nChild < nLength  and array[nChild+1]> array[nChild]) then            nChild = nChild + 1        end         --如果较大的子结点大于父结点那么把较大的子结点往上移动,替换它的父结点        if(array[i] < array[nChild]) then            swap(i,nChild,array)        else            break        end        i = nChild    endend --堆排序算法function HeapSort(array,length)    --调整序列的前半部分元素,调整完之后第一个元素是序列的最大的元素    --length/2是最后一个非叶节点,构建大根堆    for i = math.floor(length/2),1,-1 do        HeapAdjust(array,i,length)    end     --从最后一个元素开始对序列进行调整,不断的缩小调整的范围直到第一个元素   for i = length,1,-1 do        --把第一个元素和当前的最后一个元素交换        swap(i,1,array)        HeapAdjust(array,1,i - 1 )      endend arr ={9,8,7,6,5,4,3}print(table.concat(arr,","))HeapSort(arr,#arr)print(table.concat(arr,","))