LeetCode226 翻转二叉树 & 剑指Offer 27 二叉树的镜像

题目

解题

解题一：深度优先搜索

// javascript
var invertTree = function(root) {
	// 递归函数的终止条件，节点为空时返回
    if (root === null) {
        return null;
    }
    const left = invertTree(root.left);
    const right = invertTree(root.right);
    root.left = right;
    root.right = left;
    return root;
};

解题二：广度优先搜索

// javascript
var invertTree = function(root) {
    if (root === null) return null;
    // 将二叉树中的节点逐层放入队列中，再迭代处理队列中的元素
    const queue = [root];
    while (queue.length > 0) {
    	// 每次都从队列中拿一个节点，并交换这个节点的左右子树
        const node = queue.shift();
        const left = node.left;
        const right = node.right;
        node.left = right;
        node.right = left;
        // 如果当前节点的左子树不为空，则放入队列等待后续处理
        // 如果为空，没有左右子树，无须为它们交换
        if (left !== null) queue.push(left);
        // 如果当前节点的右子树不为空，则放入队列等待后续处理
        if (right !== null) queue.push(right);
    }
    // 返回处理完的根节点
    return root;
};		

时间复杂度：O(N)，其中 N 为二叉树节点的数目。我们会遍历二叉树中的每一个节点，对每个节点而言，我们在常数时间内交换其两棵子树。

空间复杂度：O(N)。使用的空间由 queue 的大小决定。