Swift数据结构和算法05_链表进阶

前言

有需要的同学可以订阅专栏:Swift数据结构和算法专题
代码地址:Swift数据结构和算法代码

正文

在本章中,我们将了解链表的五个常见场景。与大多数问题相比,这些问题相对容易,它们将有助于巩固我们对数据结构的了解。

打开启动项目开始。解决如下问题。

问题一:反转打印

创建一个以相反顺序打印链表节点的函数。例如:

1 -> 2 -> 3 -> nil

// should print out the following: 
3 
2 
1

问题 2:寻找中间节点

创建一个查找链表中间节点的函数。例如:

1 -> 2 -> 3 -> 4 -> nil 
// 中间是 3

1 -> 2 -> 3 -> nil 
// 中间是 2

问题 3:反转链表

创建一个反转链表的函数。我们可以通过操作节点来实现这一点,以便它们在另一个方向上链接。例如:

// before 
1 -> 2 -> 3 -> nil

// after 
3 -> 2 -> 1 -> nil

问题 4:合并两个列表

创建一个函数,该函数采用两个排序的链表并将它们合并为一个排序的链表。我们的目标是返回一个新的链表,其中包含按排序顺序来自两个列表的节点。我们可以假设排序顺序是升序的。例如:

// list1 
1 -> 4 -> 10 -> 11

// list2 
-1 -> 2 -> 3 -> 6

// merged list 
-1 -> 1 -> 2 -> 3 -> 4 -> 6 -> 10 -> 11

问题 5:删除所有事件

创建一个函数,从链表中删除所有出现的特定元素。该实现类似于我们为链表实现的 remove(at:) 方法。例如:

// original list 
1 -> 3 -> 3 -> 3 -> 4

// list after removing all occurrences of 3 
1 -> 4

解决方案

问题1解决方案

解决此问题的一种直接方法是使用递归。由于递归允许我们构建调用堆栈,因此我们只需在调用堆栈展开时调用打印语句。我们的第一个任务是递归遍历到最后。将以下辅助函数添加到我们的Playground

 private func printInReverse<T>(_ node: Node<T>?) { 
  // 1 
  guard let node = node else { return } 

  // 2 
  printInReverse(node.next) 
 } 
  • 1.我们首先从基本情况开始:终止递归的条件。如果 node 为 nil,则表示我们已到达列表的末尾。
    1. 这是我们的递归调用,调用与下一个节点相同的函数。

打印

我们添加打印语句的位置将决定我们是否以相反的顺序打印列表。

将函数更新为以下内容:

private func printInReverse<T>(_ node: Node<T>?) { 
  guard let node = node else { return } 
  printInReverse(node.next) 
  print(node.value) 
} 

任何代码 只有在基本情况触发之后(即,在递归函数到达列表末尾之后)才调用递归调用之后。随着递归语句的展开,节点数据被打印出来。

最后,我们需要从原始 printInReverse 函数中调用辅助方法。将其更新为如下所示:

func printInReverse<T>(_ list: LinkedList<T>) { 
  printInReverse(list.head) 
}

Playground页面的底部写下以下内容:

example(of: "printing in reverse") {
  var list = LinkedList<Int>()
  list.push(3) 
  list.push(2) 
  list.push(1)

  print("Original list: \(list)") 
  print("Printing in reverse:") 
  printInReverse(list)
}

我们应该看到以下输出:

---Example of printing in reverse---
Original list: 1 -> 2 -> 3 
Printing in reverse:
3 
2 
1

该算法的时间复杂度为 O(n),因为我们必须遍历列表的每个节点。空间复杂度同样为 O(n),因为我们隐式使用函数调用堆栈来处理每个元素。

问题 2 的解决方案

一种解决方案是让两个引用遍历列表的节点,其中一个的速度是另一个的两倍。一旦较快的参考到达终点,较慢的参考将在中间。将函数更新为以下内容:

func getMiddle<T>(_ list: LinkedList<T>) -> Node<T>? { 
  var slow = list.head 
  var fast = list.head

  while let nextFast = fast?.next { 
    fast = nextFast.next 
    slow = slow?.next 
  }
  return slow
}

while 声明中,将下一个节点绑定到 nextFast。如果有下一个节点,则快速更新到 nextFast 的下一个节点,有效地遍历列表两次。慢速指针只更新一次。这被称为跑步者的技术。

Playground页面底部写下以下内容:

example(of: "getting the middle node") {

  var list = LinkedList<Int>()

  list.push(3) 
  list.push(2) 
  list.push(1)

  print(list)

  if let middleNode = getMiddle(list) { 
    print(middleNode) 
  }
}

你应该看到以下输出:

---Example of getting the middle node---
1 -> 2 -> 3 
2 -> 3

该算法的时间复杂度为 O(n),因为我们一次遍历了列表。

问题3的解决方案

要反转链表,我们必须访问每个节点并更新下一个引用以指向另一个方向。这可能是一项棘手的任务,因为我们需要管理对多个节点的多个引用。简单的方法 我们可以使用 push方法和一个新的临时列表轻松地反转列表。更新 Playground中的代码:

extension LinkedList { 
  mutating func reverse() { 
   // 1 
  for value in self { 
  tmpList.push(value) 
  } 
  // 2 
 head = tmpList.head 
 } 
} 
    1. 首先将列表中的当前值推送到新的 tmpList。这将以相反的顺序创建一个列表。
    1. 我们将列表的头部指向反向节点。

O(n) 时间复杂度。

虽然 O(n) 是反转列表的最佳时间复杂度,但在之前的解决方案中存在大量资源成本。就像现在一样,reverse 必须为临时列表上的每个推送方法分配新节点。我们 可以避免完全使用临时列表,并通过操作每个节点的 next指针来反转列表。代码最终变得更加复杂,但我们在性能方面获得了相当大的好处。

reverse 方法更新为以下内容:

mutating func reverse() {
  tail = head 
  var prev = head 
  var current = head?.next 
  prev?.next = nil
  // more to come...
}

我们首先分配头到尾。接下来,创建两个引用——prevcurrent——来跟踪遍历。该策略相当简单:每个节点都指向列表中的下一个节点。我们将遍历列表并使每个节点都指向前一个节点:

从图中可以看出,它有点棘手。通过将 current指向 prev,我们失去了指向列表其余部分的链接。因此,我们需要管理第三个指针。在 reverse方法的底部添加以下内容:

while current != nil { 
  let next = current?.next 
  current?.next = prev 
  prev = current 
  current = next 
}

每次执行反转时,都会创建对下一个节点的新引用。在每个反转过程之后,我们将两个指针移动到接下来的两个节点。

一旦你完成了所有指针的反转,我们将把头设置到这个列表的最后一个节点。在 reverse 方法的末尾添加以下内容:

head = prev

通过在 Playground 页面底部编写以下代码来测试反向方法:

example(of: "reversing a list") {

  var list = LinkedList<Int>()

  list.push(3) 
  list.push(2) 
  list.push(1)

  print("Original list: \(list)") 
  list.reverse() 
  print("Reversed list: \(list)")
}

我们应该看到以下输出:

---Example of reversing a list---
Original list: 1 -> 2 -> 3 
Reversed list: 3 -> 2 -> 1

新的反转方法的时间复杂度仍然是 O(n),与前面讨论的简单实现相同。但是,我们不需要使用临时列表或分配新的 Node 对象,这显着提高了该算法的性能。

问题 4 解决方案

解决这个问题的方法是不断地从两个排序列表中抽取节点,并将它们添加到一个新列表中。由于这两个列表已排序,因此我们可以比较两个列表的下一个节点,看看哪一个应该是下一个添加到新列表中的节点。

我们将首先检查一个或两个列表为空的情况。将以下内容添加到 mergeSorted

guard !left.isEmpty else { 
  return right 
}

guard !right.isEmpty else { 
  return left 
}

var newHead: Node<T>?

如果一个为空,则返回另一个。我们还引入了一个新的引用来保存 Node 对象的排序列表。策略是将左右的节点按排序顺序合并到newHead上。

newHead下面写下:

// 1 
var tail: Node<T>?
var currentLeft = left.head 
var currentRight = right.head 
// 2 
if let leftNode = currentLeft, let rightNode = currentRight { 
  if leftNode.value < rightNode.value { 
    newHead = leftNode 
    currentLeft = leftNode.next 
  } else { 
    newHead = rightNode 
    currentRight = rightNode.next 
  } 
  tail = newHead
}
  1. 创建一个指向要添加到的新列表尾部的指针。这允许恒定时间的追加操作。

2.你比较left和right的第一个节点来分配newHead。

合并

接下来,我们需要遍历左右,挑选要添加的节点以确保新列表已排序。将以下内容添加到函数末尾:

// 1 
while let leftNode = currentLeft, let rightNode = currentRight { 
  // 2 
  if leftNode.value < rightNode.value { 
    tail?.next = leftNode 
    currentLeft = leftNode.next 
  } else { 
    tail?.next = rightNode 
    currentRight = rightNode.next 
  } 
  tail = tail?.next
}
  1. while 循环将继续,直到列表之一到达末尾。

  2. 很像以前,我们比较节点以找出连接到尾部的节点。

由于此循环同时依赖于 currentLeftcurrentRight,因此即使节点保留在任一列表中,它也会终止。

添加以下内容来处理剩余的节点:

if let leftNodes = currentLeft { 
  tail?.next = leftNodes 
}

if let rightNodes = currentRight { 
  tail?.next = rightNodes 
}

这会附加节点的其余部分。

总结一下,我们实例化一个新列表。无需使用 appendinsert 方法将元素插入列表,我们只需直接设置列表头部和尾部的引用:

var list = LinkedList<T>()

list.head = newHead list.tail = {

  while let next = tail?.next {

    tail = next
}

return tail 
}() 
return list

playground的底部写下以下内容:

example(of: "merging two lists") {
  var list = LinkedList<Int>()
  list.push(3) 
  list.push(2) 
  list.push(1)
  var anotherList = LinkedList<Int>()
  anotherList.push(-1) 
  anotherList.push(-2) 
  anotherList.push(-3) 
  print("First list: \(list)") 
  print("Second list: \(anotherList)") 
  let mergedList = mergeSorted(list, anotherList) 
  print("Merged list: \(mergedList)")
}

执行代码,输出结果如下:

---Example of merging two lists---
First list: 1 -> 2 -> 3 
Second list: -3 -> -2 -> -1 
Merged list: -3 -> -2 -> -1 -> 1 -> 2 -> 3

该算法的时间复杂度为 O(m + n),其中 m 是第一个列表中的节点数,n 是第二个列表中的节点数。

问题 5 的解决方案

此解决方案遍历列表,删除与要删除的元素匹配的所有节点。每次执行删除时,都需要重新连接前任节点和后继节点。虽然这可能会变得复杂,但练习这种技术是非常值得的。许多数据结构和算法将依赖于巧妙地使用指针算法来构建。

我们需要考虑几种情况。首先是清除列表前面的节点。

修剪头部

要考虑的第一种情况是列表的头部包含要删除的值。假设我们要从以下列表中删除 1:

我们希望你的新头指向 2。

remove 函数中写入以下内容:

while let head = head, head.value == value { 
  head = head?.next 
}

我们首先处理列表头部包含我们要删除的值的情况。由于可能有一系列具有相同值的节点,因此我们我们可以使用 while 循环来确保将它们全部删除。

取消链接节点

像许多与链表相关的算法一样,我们将利用我们的指针算术技能来取消链接节点。在 remove 的底部写下以下内容:

var prev = head 
var current = head?.next 
while let currentNode = current { 
  guard currentNode.value != value else { 
    prev?.next = currentNode.next 
    current = prev?.next 
    continue 
  } 
  // more to come 
}

我们需要使用两个指针遍历列表。如果需要删除节点,将触发保护语句的 else 块。

修改列表,以便绕过不想要的节点:

代码写到这里,while 语句可能永远不会终止。我们需要移动 prevcurrent 指针。在 while 循环的底部,guard语句之后编写以下内容:

prev = current 
current = current?.next

最后,我们将更新链表的尾部。当原始尾部是包含我们要删除的值的节点时,这是必要的。将以下内容添加到 removeAll的末尾:

tail = prev

Playground 页面的底部写下以下内容:

example(of: "deleting duplicate nodes") {

  var list = LinkedList<Int>()

  list.push(3) 
  list.push(2) 
  list.push(2) 
  list.push(1) 
  list.push(1)

  list.removeAll(3) 
  print(list)
}

我们将会看到如下输出结果:

---Example of deleting duplicate nodes---
1 -> 1 -> 2 -> 2

该算法的时间复杂度为 O(n),因为我们需要遍历所有元素。

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

推荐阅读更多精彩内容