用迭代法实现二叉树的中序遍历——Swift实现

最新推荐文章于 2024-06-14 02:04:20 发布

原创最新推荐文章于 2024-06-14 02:04:20 发布 · 271 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#swift #ios #算法 #leetcode

本文介绍了如何使用迭代法实现二叉树的中序遍历，以避免递归可能导致的栈溢出问题。通过创建一个栈，从根节点开始，不断将左子节点入栈直至叶子节点，然后访问节点并将其右子节点入栈，循环此过程直至栈空。同时，提供了Swift代码实现这一算法，展示了中序遍历的结果存储和遍历过程。

二叉树是一种常见的数据结构，它由节点和链接组成，每个节点包含一个值和指向其左右子节点的指针。在对二叉树进行操作时，一种常见的需求是对树进行遍历，以便访问所有节点。本文将介绍一种基于迭代法的算法来实现二叉树的中序遍历，以及如何用 Swift 语言实现这个算法。

中序遍历是指按照节点值从小到大的顺序访问二叉树的节点。通常，中序遍历使用递归实现，但在某些情况下，递归可能会导致栈溢出。为了避免这种情况，我们可以使用迭代法实现中序遍历。

算法思路

迭代法中序遍历的算法思路如下：

创建一个栈，用来存储待访问的节点。
首先将根节点入栈。
从根节点开始，不断将其左子节点入栈，直到遇到叶子节点。
弹出栈顶节点，访问该节点的值。
将栈顶节点的右子节点入栈，继续从步骤3开始执行，直到栈为空。

Swift 实现

下面是用 Swift 语言实现迭代法中序遍历的代码：

class TreeNode {
    var val: Int
    var left: TreeNode?
    var right: TreeNode?
    init(_ val: Int) {
        self.val = val
        self.left = nil
        self.right = nil
    }
}

func inorderTraversal(_ root: TreeNode?) -> [Int] {
    var result = [Int]()
    var stack = [TreeNode]()
    var current = root
    while current != nil || !stack.isEmpty {
        while current != nil {
            stack.append(current!)
            current = current?.left
        }
        current = stack.removeLast()
        result.append(current!.val)
        current = current?.right
    }
    return result
}

首先定义了一个 TreeNode 类来表示二叉树节点，包含值、左右子节点等信息。接着定义了一个 inorderTraversal 函数，接受一个根节点作为参数，返回中序遍历结果。该函数使用一个数组 result 来存储遍历结果，使用一个栈 stack 来存储待访问的节点。current 表示当前节点，初始值为根节点。

在 while 循环中，首先遍历左子树，将从当前节点开始的所有左子节点依次入栈，直到遇到叶子节点。接着弹出栈顶节点，访问其值，将值添加到结果数组中。最后遍历右子树，将右子节点设置为当前节点，继续执行 while 循环，直到栈为空。

结论

本文介绍了一种基于迭代法的算法来实现二叉树的中序遍历，以及如何用 Swift 语言实现这个算法。与递归相比，迭代法更加节省内存，并且能够避免栈溢出的问题。迭代法中序遍历算法的时间复杂度为 O(n)，其中 n 是二叉树节点数，空间复杂度为 O(h)，其中 h 是二叉树高度。在实际应用中，我们可以根据具体情况选择递归或迭代法实现二叉树遍历，以满足不同的需求。