226. 翻转二叉树

翻转一棵二叉树。
示例:
输入:
  4
  /  \  
 2  7
 /  \   /  \  
1  3 6  9
输出:

  4
  / \  
 7  2
 /  \  /  \  
9 6 3 1
备注:
这个问题是受到 Max Howell 的 原问题 启发的 :

谷歌:我们90%的工程师使用您编写的软件(Homebrew),但是您却无法在面试时在白板上写出翻转二叉树这道题,这太糟糕了。

解题思路:此题很简单,根据根节点左右分开,之后通过递归左右交换即可

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode invertTree(TreeNode root) {
        if(root == null)
            return root;
        TreeNode node = new TreeNode(0);
        node = root.left;
        root.left = root.right;
        root.right = node;
        invertTree(root.left);
        invertTree(root.right);
        return root;
    }
}

更简洁的写法:

public TreeNode invertTree(TreeNode root) {
    if (root == null) {
        return null;
    }
    TreeNode right = invertTree(root.right);
    TreeNode left = invertTree(root.left);
    root.left = right;
    root.right = left;
    return root;
}

复杂度分析

既然树中的每个节点都只被访问一次,那么时间复杂度就是 O(n),其中 n 是树中节点的个数。在反转之前,不论怎样我们至少都得访问每个节点至少一次,因此这个问题无法做地比 O(n) 更好了。

本方法使用了递归,在最坏情况下栈内需要存放 O(h)个方法调用,其中 h是树的高度。由于 h∈O(n),可得出空间复杂度为 O(n)。
leetcode解题思路:迭代法
方法二 (迭代) 【通过】
我们也可以用迭代方法来解决这个问题,这种做法和深度优先搜索(Breadth-fist Search, BFS)很接近。

算法

这个方法的思路就是,我们需要交换树中所有节点的左孩子和右孩子。因此可以创一个队列来存储所有左孩子和右孩子还没有被交换过的节点。开始的时候,只有根节点在这个队列里面。只要这个队列不空,就一直从队列中出队节点,然后互换这个节点的左右孩子节点,接着再把孩子节点入队到队列,对于其中的空节点不需要加入队列。最终队列一定会空,这时候所有节点的孩子节点都被互换过了,直接返回最初的根节点就可以了。

public TreeNode invertTree(TreeNode root) {
    if (root == null) return null;
    Queue queue = new LinkedList();
    queue.add(root);
    while (!queue.isEmpty()) {
        TreeNode current = queue.poll();
        TreeNode temp = current.left;
        current.left = current.right;
        current.right = temp;
        if (current.left != null) queue.add(current.left);
        if (current.right != null) queue.add(current.right);
    }
    return root;
}

复杂度分析
既然树中的每个节点都只被访问/入队一次,时间复杂度就是 O(n),其中 n 是树中节点的个数。
空间复杂度是 O(n),即使在最坏的情况下,也就是队列里包含了树中所有的节点。对于一颗完整二叉树来说,叶子节点那一层拥有 n/2(向上取整))=O(n) 个节点。

你可能感兴趣的:(226. 翻转二叉树)