LeetCode 226. 翻转二叉树 *

翻转一棵二叉树。

测试用例

示例：
输入：

输出：

解题思路

递归

这是一道很经典的二叉树问题。显然，我们从根节点开始，递归地对树进行遍历，并从叶子结点先开始翻转。如果当前遍历到的节点 $root$ 的左右两棵子树都已经翻转，那么我们只需要交换两棵子树的位置，即可完成以 $root$ 为根节点的整棵子树的翻转。

Code

/**
 * 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 null;
        }
        TreeNode node = root.left;
        root.left = invertTree(root.right);
        root.right = invertTree(node);
        return root;
    }    
}

class Solution {
	//自顶向下
    public TreeNode invertTree(TreeNode root) {
        invert(root);
        return root;    
    }
    public void invert(TreeNode node){
        if(node == null){
            return;
        }
        TreeNode temp = node.left;
        node.left = node.right;
        node.right = temp;
        invert(node.left);
        invert(node.right);
    }
}

杂度分析

• 时间复杂度：$O(N)$，其中 N 为二叉树节点的数目。我们会遍历二叉树中的每一个节点，对每个节点而言，我们在常数时间内交换其两棵子树。
• 空间复杂度：$O(N)$。使用的空间由递归栈的深度决定，它等于当前节点在二叉树中的高度。在平均情况下，二叉树的高度与节点个数为对数关系，即 $O(logN)$。而在最坏情况下，树形成链状，空间复杂度为 $O(N)$。