每日一题 --- 二叉树的遍历[力扣][Go]

type TreeNode struct {
  Val int
  Left *TreeNode
  Right *TreeNode
}

// 二叉树的先序遍历
func preorderTraversal(root *TreeNode) []int {
  var ans []int
  var dfs func(*TreeNode)
  dfs = func(t *TreeNode) {
    if t == nil {
      return
    }
    ans = append(ans, t.Val)
    dfs(t.Left)
    dfs(t.Right)
  }
  dfs(root)
  return ans
}

// 二叉树的中序遍历
func inorderTraversal(root *TreeNode) []int {
  var ans []int
  var dfs func(*TreeNode)
  dfs = func(t *TreeNode) {
    if t == nil {
      return
    }
    dfs(t.Left)
    ans = append(ans, t.Val)
    dfs(t.Right)
  }
  dfs(root)
  return ans
}

// 二叉树的后序遍历
func postorderTraversal(root *TreeNode) []int {
  var ans []int
  var dfs func(*TreeNode)
  dfs = func(t *TreeNode) {
    if t == nil {
      return
    }
    dfs(t.Left)
    dfs(t.Right)
    ans = append(ans, t.Val)
  }
  dfs(root)
  return ans
}