代码随想录算法训练营第十五天（二）|226.翻转二叉树

最新推荐文章于 2024-06-12 10:00:23 发布

张小凡不凡

最新推荐文章于 2024-06-12 10:00:23 发布

阅读量324

点赞数 9

分类专栏：代码随想录算法训练营文章标签：算法数据结构

本文链接：https://blog.csdn.net/qq_33350950/article/details/136574709

版权

代码随想录算法训练营专栏收录该内容

33 篇文章 0 订阅

订阅专栏

226.翻转二叉树

题目

翻转一棵二叉树。

在这里插入图片描述

解题思路

其实就是循环将根节点的左右子节点互换就行
可以用层序遍历，每遍历一层就互换其左右子节点，符合惯性思维
也可以用递归法，前序遍历和后序遍历

代码

层序遍历

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        while (!que.empty()) {
            int size = que.size();
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                swap(node->left, node->right); // 节点处理
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
        }
        return root;
    }
};

递归算法

确定递归函数的参数和返回值
本例子中，就是互换node节点的left和right，那么每次传入node即可
```
TreeNode* invertTree(TreeNode* root)
```
确定终止条件
如果当前节点为空，则返回该节点
```
if(root==nullptr) return root;
```

确定循环主体
循环互换其左子树和右子树上的左右子节点即可

swap(root->left, root->right);
invertTree(root->left);
invertTree(root->right);

完整代码如下：

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if (root == NULL) return root;
        swap(root->left, root->right);  // 中
        invertTree(root->left);         // 左
        invertTree(root->right);        // 右
        return root;
    }
};