算法笔记
之前一直在使用层序遍历解决二叉树类的题目,加深对于层序遍历的理解,今天的题目同样可以使用层序遍历解决,但主要还是讲解一下递归的做法。
题目描述
解题思路
使用递归解决二叉树的题目依然可以用三部曲来进行,先确定参数返回值,然后设定终止条件,最后确定递归的逻辑。先考虑本题应该使用哪种遍历方法,翻转二叉树如下图所示,根据图片就可以更容易理解使用哪种遍历更方便,反转只需要使用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;
}
};
结语
翻转二叉树是一道不算难的题目,但对于我们来了解遍历方法是很有用的,本题主要是培养拿到相关题目首先要确认是前中后遍历的思维方式。