二叉树遍历

package main

import "fmt"

// 定义二叉树节点结构体
type TreeNode struct {
    Val   any
    Left  *TreeNode
    Right *TreeNode
}

// 前序遍历
func preOrderTraversal(root *TreeNode) {
    if root == nil {
        return
    }
    fmt.Printf("%v ", root.Val)
    preOrderTraversal(root.Left)
    preOrderTraversal(root.Right)
}

// 中序遍历
func inOrderTraversal(root *TreeNode) {
    if root == nil {
        return
    }
    inOrderTraversal(root.Left)
    fmt.Printf("%v ", root.Val)
    inOrderTraversal(root.Right)
}

// 后序遍历
func postOrderTraversal(root *TreeNode) {
    if root == nil {
        return
    }
    postOrderTraversal(root.Left)
    postOrderTraversal(root.Right)
    fmt.Printf("%v ", root.Val)
}

func main() {
    // 创建一个示例二叉树
    /*
            1
           / \
          2   3
         / \
        4   5
    */
    root := &TreeNode{Val: 1}
    root.Left = &TreeNode{Val: 2}
    root.Right = &TreeNode{Val: 3}
    root.Left.Left = &TreeNode{Val: 4}
    root.Left.Right = &TreeNode{Val: 5}

    // 前序遍历
    fmt.Print("Pre-order Traversal: ")
    preOrderTraversal(root)
    fmt.Println()

    // 中序遍历
    fmt.Print("In-order Traversal: ")
    inOrderTraversal(root)
    fmt.Println()

    // 后序遍历
    fmt.Print("Post-order Traversal: ")
    postOrderTraversal(root)
    fmt.Println()
}