swift 二叉树三种(前序,中序,后序)遍历方式

class TreeNode: NSObject {
    var val: Int = 0
    var left: TreeNode?
    var right: TreeNode?
    override init() {}
    
    init(val: Int, left: TreeNode?, right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
}

class BinaryTree {
    ///二叉树的前序遍历
    ///给定二叉树的根节点 root ,返回它节点值的 前序 遍历
    func preorderTraversal(_ root: TreeNode?) -> [Int] {
        guard let root = root else { return [] }
        var list: [Int] = []
        list.append(root.val)
        list += preorderTraversal(root.left)
        list += preorderTraversal(root.right)
        return list
    }
    ///二叉树的中序遍历
    ///给定一个二叉树的根节点 root ,返回它的 中序 遍历
    func inorderTraversal(_ root: TreeNode?) -> [Int] {
        guard let root = root else { return [] }
        var list: [Int] = []
        list += inorderTraversal(root.left)
        list.append(root.val)
        list += inorderTraversal(root.right)
        return list
    }
    ///二叉树的后序遍历
    ///给定一个二叉树的根节点 root,返回它的 后序 遍历
    func postorderTraversal(_ root: TreeNode?) -> [Int] {
        guard let root = root else { return [] }
        var list: [Int] = []
        list += postorderTraversal(root.left)
        list += postorderTraversal(root.right)
        list.append(root.val)
        return list
    }
    
}
//root = [1,null,2,3]
let binary = BinaryTree()
let b = TreeNode(val: 3, left: nil, right: nil)
let a = TreeNode(val: 2, left: b, right: nil)
let root = TreeNode(val: 1, left: nil, right: a)
binary.preorderTraversal(root)
binary.inorderTraversal(root)
binary.postorderTraversal(root)

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值