LeetCode 226 Invert Binary Tree（翻转二叉树）

最新推荐文章于 2023-02-20 18:30:00 发布

nudt_oys

最新推荐文章于 2023-02-20 18:30:00 发布

阅读量306

点赞数

分类专栏：数据结构 LeetCode题解——C/C++版文章标签： LeetCode

本文链接：https://blog.csdn.net/qq_26658823/article/details/70337118

版权

LeetCode题解——C/C++版同时被 2 个专栏收录

69 篇文章 13 订阅

订阅专栏

数据结构

66 篇文章 0 订阅

订阅专栏

Invert a binary tree.

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

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

题目大意：翻转二叉树。

解题思路：递归。不过同样是递归，感觉自己写的代码没有别人写的优雅……

代码如下：

//自己写的
 void swap(struct TreeNode** a, struct TreeNode** b)
 {
     struct TreeNode* tmp = *a;
     *a = *b;
     *b = tmp;
 }
 
struct TreeNode* invertTree(struct TreeNode* root) {
    if(root == NULL) return NULL;
    if(root->right && root->left){
        root->right = invertTree(root->right);
        root->left = invertTree(root->left);
        swap(&(root->right), &(root->left));
        
    }else if(root->right){
        root->right = invertTree(root->right);
        swap(&(root->right), &(root->left));
        
    }else if(root->left){
        root->left = invertTree(root->left);
        swap(&(root->right), &(root->left));
    }
}

//别人写的
struct TreeNode* invertTree(struct TreeNode* root) {
    if(root == NULL) return NULL;
    struct TreeNode *tmp = root->left;
    root->left = invertTree(root->right);
    root->right = invertTree(tmp);
    return root;
}