【数据结构与算法 - Swift实现】12 - 时间复杂度为O(n^2)的排序算法

时间复杂度为O(n^2)的排序算法性能并不好,但作为刚入门算法的同学来说,还是有必要学习一下的。这里我们会讲到三种排序算法:1)冒泡排序;2)选择排序;3)插入排序。

冒泡排序

原理

首先介绍下冒泡排序算法的原理:

  • 从第一个元素开始,比较相邻的元素,如果第一个比第二个大,就交换它们的值
  • 对每一组相邻元素做同样的操作,完成最后一组的操作后,最后的元素肯定是最大的数
  • 除了最后已排好序的元素,对其他所有的元素重复以上的步骤
  • 重复上面的步骤,直到没有任何一组数字需要比较

例子

下面看一个例子,假设我们有下面的数组:

    +--------+    +--------+    +--------+    +--------+
    |        |    |        |    |        |    |        |
    |    8   |    |    7   |    |   14   |    |    5   |
    |        |    |        |    |        |    |        |
    +--------+    +--------+    +--------+    +--------+

根据上面的原理,首先将第一个和第二个比较,87 交换,结果如下:

    +--------+    +--------+    +--------+    +--------+
    |        |    |        |    |        |    |        |
    |    7   |    |    8   |    |   14   |    |    5   |
    |        |    |        |    |        |    |        |
    +--------+    +--------+    +--------+    +--------+

第二个和第三个继续比较,第三个数 14 大于第二个数 8,无需交换。

第三个和第四个比较,145 交换,至此第一轮的比较已经完成,最后一个数就是最大的数。结果如下:

    +--------+    +--------+    +--------+    +--------+
    |        |    |        |    |        |    |        |
    |    7   |    |    8   |    |    5   |    |   14   |
    |        |    |        |    |        |    |        |
    +--------+    +--------+    +--------+    +--------+

下面是第二轮对比,第一组 78 的顺序已经是对的,无需再交换;第一组 85,需要交换,结果如下:

    +--------+    +--------+    +--------+    +--------+
    |        |    |        |    |        |    |        |
    |    7   |    |    5   |    |    8   |    |   14   |
    |        |    |        |    |        |    |        |
    +--------+    +--------+    +--------+    +--------+

最后一个已经是做大的了,不用再对比。也就是说每经过一轮对比,下一轮要对比的元素就会少一个。

下面第三轮对比,75 交换,结果如下:

    +--------+    +--------+    +--------+    +--------+
    |        |    |        |    |        |    |        |
    |    5   |    |    7   |    |    8   |    |   14   |
    |        |    |        |    |        |    |        |
    +--------+    +--------+    +--------+    +--------+

因为这是第三轮的对比,后面两个数不需要再对比,到此,排序完成。

实现

为了保证方法的通用性,我们不仅对数组进行排序。Swift 中有一个协议叫做 MutableCollection, 这个协议规定 collection 必须支持通过下标去访问或者修改里面的元素,所以我们直接对 MutableCollection 类型的对象进行排序。

实现代码如下:

func bubbleSort<T>(_ collection: inout T)
    where T: MutableCollection, T.Element: Comparable {
        guard collection.count > 1 else {
            return
        }
        for end in collection.indices.reversed() {
            var swapped = false
            var current = collection.startIndex
            while current < end {
                let next = collection.index(after: current)
                if collection[next] < collection[current] {
                    collection.swapAt(next, current)
                    swapped = true
                }
                current = next
            }
            if !swapped {
                return
            }
        }
}

代码解析:

  • 首先,至少要有两个元素,我们才进行排序,如果 collection.count 不大于 1,直接返回
  • 因为每经过一轮对比,下一轮对比就要忽略最后的元素,所以 for 循环的循环条件,把 indices 倒序
  • 定义了一个 swapped 变量,记录当前这轮对比是否有调换过元素
  • while 循环执行当前这一轮对比,如果后面的元素小于前面的元素就调换位置
  • 如果 swappedfalse,说明当前这轮对比没有调换过元素,也就意味着已经排好序,提前退出循环

测试

var ints = [4,5,5467,73,234,678,87,989]
bubbleSort(&ints)
print(ints)

// 结果
[4, 5, 73, 87, 234, 678, 989, 5467]

结果正确。

选择排序

选择排序对冒泡排序进行了一些改进,减少了元素之间的交换次数。选择排序只在每一轮对比完成之后才叫换一次。

例子

下面看一个例子,假设我们有下面的数组:

    +--------+    +--------+    +--------+    +--------+
    |        |    |        |    |        |    |        |
    |    8   |    |    7   |    |   14   |    |    5   |
    |        |    |        |    |        |    |        |
    +--------+    +--------+    +--------+    +--------+

首先找到最小的元素 585 交换,结果如下:

    +--------+    +--------+    +--------+    +--------+
    |        |    |        |    |        |    |        |
    |    5   |    |    7   |    |   14   |    |    8   |
    |        |    |        |    |        |    |        |
    +--------+    +--------+    +--------+    +--------+

再找到除第一个元素外的最小元素 77 在第二个位置,无需交换。

再找到除前面两个元素外的最小元素8,放到第三个位置,结果如下:

    +--------+    +--------+    +--------+    +--------+
    |        |    |        |    |        |    |        |
    |    5   |    |    7   |    |    8   |    |   14   |
    |        |    |        |    |        |    |        |
    +--------+    +--------+    +--------+    +--------+

到此,排序完成。很明显,交换的次数比冒泡排序少了很多。

实现

为了保证方法的通用性,跟冒泡排序一样,我们直接对 MutableCollection 类型的对象进行排序。

func selectionSort<T>(_ collection: inout T)
    where T: MutableCollection, T.Element: Comparable {
        guard collection.count > 1 else {
            return
        }
        for current in collection.indices {
            var lowest = current
            var next = collection.index(after: current)
            while next < collection.endIndex {
                if collection[next] < collection[lowest] {
                    lowest = next
                }
                next = collection.index(after: next)
            }
            if lowest != current {
                collection.swapAt(lowest, current)
            }
        }
}

代码解析:

  • 首先,至少要有两个元素,我们才进行排序,如果 collection.count 不大于 1,直接返回
  • 因为每经过一轮交换,下一轮要忽略前面的元素,所以使用 for 循环,条件是从最小的下标开始
  • while 循环找出最小的元素
  • 如果找到的最小元素与当前元素不相等,则交换位置

测试

var ints = [4,5,5467,73,234,678,87,989]
selectionSort(&ints)
print(ints)

// 结果
[4, 5, 73, 87, 234, 678, 989, 5467]

结果正确。

插入排序

插入排序的原理:将当前的元素与上一个进行对比,如果当前的元素小于上一个元素,则交换位置,重复这个步骤,直到把它放到合适的位置为止。

例子

下面看一个例子,假设我们有下面的数组:

    +--------+    +--------+    +--------+    +--------+
    |        |    |        |    |        |    |        |
    |    8   |    |    7   |    |   14   |    |    5   |
    |        |    |        |    |        |    |        |
    +--------+    +--------+    +--------+    +--------+

因为第一个元素前面没有其他元素,所以忽略第一个元素。

第二个元素 7 与前面的 8 对比,需要交换位置,结果如下:

    +--------+    +--------+    +--------+    +--------+
    |        |    |        |    |        |    |        |
    |    7   |    |    8   |    |   14   |    |    5   |
    |        |    |        |    |        |    |        |
    +--------+    +--------+    +--------+    +--------+

第三个元素 14 比前面的两个元素都大,无需交换。

第四个元素 5,比前面的元素都小,所以他会一直与前面的元素交换,直到它在第一个位置。结果如下:

    +--------+    +--------+    +--------+    +--------+
    |        |    |        |    |        |    |        |
    |    5   |    |    7   |    |    8   |    |   14   |
    |        |    |        |    |        |    |        |
    +--------+    +--------+    +--------+    +--------+

实现

为了保证方法的通用性,我们不直接对数组类型进行排序,而是 Swift 的集合类型。首先我们要交换元素的位置,集合类型必须是 MutableCollection 类型;另外我们还要访问上一个元素,BidirectionalCollection类型才拥有这个特性。所以我们排序的类型是 BidirectionalCollection & MutableCollection。代码如下:

func insertionSort<T>(_ collection: inout T)
    where T: BidirectionalCollection & MutableCollection,
    T.Element: Comparable {
        guard collection.count > 1 else {
            return
        }
        for current in collection.indices {
            var moving = current
            while moving > collection.startIndex {
                let previous = collection.index(before: moving)
                if collection[previous] > collection[moving] {
                    collection.swapAt(previous, moving)
                } else {
                    break
                }
                moving = previous
            }
        }
}
  • 首先,至少要有两个元素,我们才进行排序,如果 collection.count 不大于 1,直接返回
  • for 循环保证当前将要往左运动的元素
  • while 循环里面,把当前元素与上一个元素对比,如果上一个元素大于当前元素,则交换位置;否则前面的元素肯定已经排好序,所以直接 break

测试

var ints = [4,5,5467,73,234,678,87,989]
insertionSort(&ints)
print(ints)

// 结果
[4, 5, 73, 87, 234, 678, 989, 5467]

结果正确。

完整代码 >>

参考资料

Data Structures and Algorithms in Swift --- raywenderlich.com,如果想看原版书籍,请点击链接购买。

欢迎加入我管理的Swift开发群:536353151

下一篇文章:【数据结构与算法 - Swift实现】13 - 归并排序 (Merge Sort)

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,294评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,493评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,790评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,595评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,718评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,906评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,053评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,797评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,250评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,570评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,711评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,388评论 4 332
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,018评论 3 316
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,796评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,023评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,461评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,595评论 2 350