Swift实现二叉树的遍历（递归和非递归）

悠悠悠哉e

已于 2022-03-22 16:50:37 修改

阅读量3k

点赞数

分类专栏： Swift 文章标签： swift

于 2022-03-22 10:16:01 首次发布

本文链接：https://blog.csdn.net/zjh467545737/article/details/123653234

版权

本文详细介绍了如何使用Swift编程语言实现二叉树的递归和非递归遍历，包括前序、中序、后序遍历，以及广度优先和深度优先搜索遍历。同时，还给出了使用数组构造二叉树的方法。

摘要由CSDN通过智能技术生成

1、创建一个TreeNode的Class

public class TreeNode {
    var val: Int
    var left: TreeNode?
    var right: TreeNode?
    
    init() {
        self.val = 0; self.left = nil; self.right = nil
    }
    init(_ val:Int) {
        self.val = val; self.left = nil; self.right = nil
    }
    
    public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
    
// 这里复写了 == 比较方法，用于非递归后序遍历使用。主要是通过内存地址来判断是否为同一个节点
    static func == (a: TreeNode, b: TreeNode) -> Bool {
       return a.getAnyObjectMemoryAddress(object: a) == b.getAnyObjectMemoryAddress(object: b)
    }
    
    // 取出某个对象的地址
      func getAnyObjectMemoryAddress(object: AnyObject) -> String {
          let str = Unmanaged<AnyObject>.passUnretained(object).toOpaque()
          return String(describing: str)
      }
}

2、递归实现前、中、后序的遍历

代码量比较少，也比较简单

// 前序
func fristRootTree(_ root: TreeNode?) {
    if root == nil { return }
    print("\(root!.val) -> ")
    fristRootTree(root!.left)
    fristRootTree(root!.right)
}

// 中序
func middleRootTree(_ root: TreeNode?) {
    if root == nil { return }
    middleRootTree(root!.left)
    print("\(root!.val) -> ")
    middleRootTree(root!.right)
}

// 后序
func lastRootTree(_ root: TreeNode?) {
    if root == nil { return }
    lastRootTree(root?.left)
    lastRootTree(root?.right)
    print("\(root!.val) -> ")
}