Fork me on GitHub

leetcode之226. 翻转二叉树

发表于 | 分类于
字数统计 313 字 | 阅读时长 1 分钟

题目描述：

翻转一棵二叉树。

示例：

输入：

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

输出：

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

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

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

解题思路一：

时间复杂度：$O(n)$, 空间复杂度:$O(1)$.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        
        if(root == NULL) return root;

        TreeNode *left = invertTree(root->left);
        TreeNode *right = invertTree(root->right);
        
        root->left = right;
        root->right = left;
        
        return root;
    }    
};

解题思路二：

时间复杂度：$O(n)$, 空间复杂度:$O(1)$.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* invertTree(TreeNode* root) 
   {
        if (!root) return nullptr;
        // 交换两个指针，是交换指针而不是值
        auto *pTmp = root->left;
        root->left = root->right;
        root->right = pTmp;
        // 递归的计算左子树
        invertTree(root->left);
        // 递归的计算右子树
        invertTree(root->right);
        return root;
   }    
};