精选力扣500题 第58题 LeetCode 226. 翻转二叉树c++/java详细题解

Posted 2021-08-05 林深时不见鹿

目录

• • • 1、题目
    • 2、思路
    • 3、c++代码
    • 4、java代码

1、题目

翻转一棵二叉树。

示例：

输入：

 	 4
   /   \\
  2     7
 / \\   / \\
1   3 6   9

输出：

	  4
    /   \\
  7     2
 / \\   / \\
9   6 3   1

备注:
这个问题是受到 Max Howell 的 原问题 启发的 ：

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

2、思路

(递归) $O(n)$

我们可以发现翻转后的树就是将原树的所有节点的左右儿子互换！

所以我们递归遍历原树的所有节点，将每个节点的左右儿子互换即可。

时间复杂度分析： 每个节点仅被遍历一次，所以时间复杂度是 $O(n)$。

3、c++代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if( !root) return nullptr;
        swap(root->left,root->right); //互换左右儿子
        invertTree(root->left); //递归左子树
        invertTree(root->right); //递归右子树
        return root;
    }
};

4、java代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode invertTree(TreeNode root) {
        if( root == null) return null;
        TreeNode temp = root.left; //互换左右儿子
        root.left = root.right;
        root.right = temp;
        invertTree(root.left); //递归左子树
        invertTree(root.right); //递归右子树
        return root;
    }
}

原题链接： 226. 翻转二叉树