一、题目内容 简单

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

示例 1:

1. 翻转二叉树(226) - 图1 输入:root = [4,2,7,1,3,6,9] 输出:[4,7,2,9,6,3,1]

示例 2:

1. 翻转二叉树(226) - 图2 输入:root = [2,1,3] 输出:[2,3,1]

示例 3:

输入:root = [] 输出:[]

提示:

  • 树中节点数目范围在 [0, 100]
  • -100 <= Node.val <= 100

    二、解题思路

    递归法:递归到最底层,从最底层开始,反转 node 的 node.leftnode.right

递归法的弊端:每层递归,都会压一个函数栈,性能不好。

三、具体代码

  1. /**
  2. * @param {TreeNode} root
  3. * @return {TreeNode}
  4. */
  5. var invertTree = function (root) {
  6. if (!root) return null;
  7. return new TreeNode(root.val, invertTree(root.right), invertTree(root.left));
  8. };
  9. /**
  10. * 时间复杂度:O(n)
  11. * 空间复杂度:O(n)
  12. */

四、其他解法

迭代法

深度优先遍历1

如果我们这样翻转应该还可以这样实现。从根节点开始,翻转其左右的子树。接着进入左右子树 A 和 B。
以 A 为根节点,翻转 A 的子树,同理翻转 B 的子树。然后一层层得翻转。

按照这个思路,我们应该声明一个栈,把 A 节点 的左右子节点,压入栈,目的是等待下一次拿出这两个节点,翻转他们各自的左右子树。
我们把 A 节点的左右子树翻转,然后继续循环,直到栈为空,说明翻转完成。

  1. /**
  2. * @param {TreeNode} root
  3. * @return {TreeNode}
  4. */
  5. var invertTree = function (root) {
  6. if (!root) return null;
  7. const stack = [root];
  8. const invertNode = (root, left, right) => {
  9. root.left = right;
  10. root.right = left;
  11. }
  12. while (stack.length) {
  13. const node = stack.pop();
  14. node.right && stack.push(node.right);
  15. node.left && stack.push(node.left);
  16. invertNode(node, node.left, node.right);
  17. }
  18. return root;
  19. };
  20. /**
  21. * 时间复杂度:O(n)
  22. * 空间复杂度:O(n)
  23. */

深度优先遍历2

  1. /**
  2. * Definition for a binary tree node.
  3. * function TreeNode(val, left, right) {
  4. * this.val = (val===undefined ? 0 : val)
  5. * this.left = (left===undefined ? null : left)
  6. * this.right = (right===undefined ? null : right)
  7. * }
  8. */
  9. /**
  10. * @param {TreeNode} root
  11. * @return {TreeNode}
  12. */
  13. var invertTree = function (root) {
  14. if (!root) return null
  15. const queue = [root]
  16. while (queue.length) {
  17. const node = queue.shift()
  18. if (node.left) queue.push(node.left)
  19. if (node.right) queue.push(node.right)
  20. const tmp = node.left
  21. node.left = node.right
  22. node.right = tmp
  23. }
  24. return root
  25. };