leetcode 226 翻转二叉树

题目

给你一棵二叉树的根节点 root ，翻转这棵二叉树，并返回其根节点

示例

输入：root = [4,2,7,1,3,6,9]
输出：[4,7,2,9,6,3,1]

解析

首先这道题，要求返回的是根节点，（之前的前中后序遍历返回的都是一个新数组）；这道题翻转的话，遍历交换左右子节点就可以（前序、后序、层次都可以）
前序递归法：

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func invertTree(root *TreeNode) *TreeNode {
    if root == nil {
        return nil
    }
    invertTree(root.Left)
    invertTree(root.Right)
    root.Left, root.Right = root.Right, root.Left
    return root
}

---

前序迭代法(基本和之前的迭代法一样)：

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func invertTree(root *TreeNode) *TreeNode {
    if root == nil {
        return nil
    }
    st := list.New()
    st.PushBack(root)
    for st.Len() > 0 {
        node := st.Remove(st.Back()).(*TreeNode)
        node.Left, node.Right = node.Right, node.Left
        if node.Right != nil {
            st.PushBack(node.Right)
        }
        if node.Left != nil {
            st.PushBack(node.Left)
        }
    }
    return root
}

---

层次遍历：

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func invertTree(root *TreeNode) *TreeNode {
    if root == nil {
        return nil
    }
    queue := list.New()
    queue.PushBack(root)
    for queue.Len() > 0 {
        length := queue.Len()
        for i:=0; i < length; i++ {
            node := queue.Remove(queue.Front()).(*TreeNode)
            node.Left, node.Right = node.Right, node.Left
            if node.Left != nil {
                queue.PushBack(node.Left)
            }
            if node.Right != nil {
                queue.PushBack(node.Right)
            }
        }
    }
    return root
}

---

总结：这道题其实用啥方法都不太重要，感觉只要能遍历走完这棵树，有左右节点的交换一下，最后就可以了