zlatan的算法笔记14--翻转二叉树（力扣226题）

算法笔记

之前一直在使用层序遍历解决二叉树类的题目，加深对于层序遍历的理解，今天的题目同样可以使用层序遍历解决，但主要还是讲解一下递归的做法。

题目描述

解题思路

使用递归解决二叉树的题目依然可以用三部曲来进行，先确定参数返回值，然后设定终止条件，最后确定递归的逻辑。先考虑本题应该使用哪种遍历方法，翻转二叉树如下图所示，根据图片就可以更容易理解使用哪种遍历更方便，反转只需要使用swap函数来交换左右子树即可，比较容易，这样无论是使用中序遍历还是后序遍历都可以很好实现，但是使用中序遍历则会反转两遍就会比较麻烦，本题主要还是要注意遍历方法。

代码部分

递归做法

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if(root == nullptr) {
            return nullptr;
        }
        swap(root->left, root->right);
        invertTree(root->left);
        invertTree(root->right);
        return root;
    }
};

层序做法

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if(root == nullptr) {
            return root;
        }
        stack<TreeNode*> st;
        st.push(root);
        while(!st.empty()) {
            TreeNode* node = st.top();
            st.pop();
            swap(node -> left, node -> right);
            if(node -> right) {
                st.push(node -> right);
            }
            if(node -> left) {
                st.push(node -> left);
            }
        }
        return root;
    }
};

结语

翻转二叉树是一道不算难的题目，但对于我们来了解遍历方法是很有用的，本题主要是培养拿到相关题目首先要确认是前中后遍历的思维方式。