swift 文章收集swiftiOS学习记录

Swift之堆排序

2016-08-06  本文已影响126人  我系哆啦

在介绍对排序之前,先介绍“树”的相关概念!
树:指不包含回路的连通无向图。有以下一些特性:

二叉树

二叉树是一种特使的树,二叉树的特点是每个节点最多只有两个子节点,左边的叫做左子树,右边的叫做右儿树。二叉树中还有两种特殊的树,满二叉树和完全二叉树,满二叉树是指二叉树中每个内部节点都有两个字节点(严格定义是一颗深度为h且有2的h次方-1个节点的二叉树);完成二叉树是指除了最右边位置上有一个或者几个叶节点缺少外,其他都是丰满(严格定义是:若二叉树的高度为h,除第h层外,其他各层(1~h-1)的节点数都达到最大个数,第h层从右向左连续缺若干节点)

堆是一种特使的完全二叉树,堆分为两种。一种是所有父节点都要比子节点大,这称为最大堆;另一种是素有的父节点比所有的子节点都要小,这称为最小堆。如下图所示,是一个最小堆。


最小堆.png

堆排序

堆的特性非常适合来用来做排序算法。下面介绍一种利用堆来排序的算法:

完整代码如下:

<pre>
var tempList:[Int] = [99,5,36,7,22,17,46,12,2,19,25,28,1,92]

//交换数组内的元素
func swapInTemp(x:Int,y:Int) {
let t:Int = tempList[x]
tempList[x] = tempList[y]
tempList[y] = t
}

//向下调整
func siftDown(i:Int,count:Int) {

var flag = 0
var temp = 0
var tempI = i

while (flag == 0) && (tempI*2 < (count - 1)) {
    
    if tempList[tempI] < tempList[tempI*2 + 1] { //小于左子树
        temp = tempI*2 + 1
    } else {
        temp = tempI
    }
    
    if (tempI*2 + 2) <= (count - 1) { //如果有右子树
        if tempList[temp] < tempList[tempI*2 + 2] {
            temp = tempI*2 + 2
        }
    }
    
    if temp != tempI {
        swapInTemp(x: temp, y: tempI)
        tempI = temp
    }
    else{
        flag = 1
    }
}

}

//创建最大堆
func creatheap(list: inout [Int]) {

var i = list.count / 2 - 1

while i >= 0 {
    siftDown(i: i, count: list.count)
    i-=1
}

}

//堆排序(从小到大)
func heapSort(list: inout [Int]) {

var max = list.index(before: list.endIndex)
while max > 0 {
    swapInTemp(x: 0, y: max)
    siftDown(i: 0, count: max)
    max-=1
}

}

creatheap(list: &tempList)
heapSort(list: &tempList)
print(tempList)//[1, 2, 5, 7, 12, 17, 19, 22, 25, 28, 36, 46, 92, 99]
</pre>

上一篇下一篇

猜你喜欢

热点阅读