Welcome to you, 每日一刷系列
翻转二叉树
可以发现想要翻转它,其实就把每一个节点的左右孩子交换一下就可以了。
关键在于遍历顺序,前中后序应该选哪一种遍历顺序? (一些同学这道题都过了,但是不知道自己用的是什么顺序)
遍历的过程中去翻转每一个节点的左右孩子就可以达到整体翻转的效果。
注意只要把每一个节点的左右孩子翻转一下,就可以达到整体翻转的效果
这道题目使用前序遍历和后序遍历都可以,唯独中序遍历不方便,因为中序遍历会把某些节点的左右孩子翻转了两次!建议拿纸画一画,就理解了
那么层序遍历可以不可以呢?依然可以的!只要把每一个节点的左右孩子翻转一下的遍历方式都是可以的!
递归
前序遍历
把交换的操作,放在递归子树之前。问题是在压栈前被解决
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if(root==nullptr)
{
return 0;
}
swap(root->left,root->right);//交换
invertTree(root->left);//左
invertTree(root->right);//右
return root;
}
};
后序遍历
先交换左右子树。那么递归就会先压栈压到底。然后才做交换。
问题是在递归出栈时解决的。
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if(root==nullptr)
{
return 0;
}
invertTree(root->left);
invertTree(root->right);
swap(root->left,root->right);
return root;
}
};
中序遍历
文中我指的是递归的中序遍历是不行的,因为使用递归的中序遍历,某些节点的左右孩子会翻转两次。
如果非要使用递归中序的方式写,也可以,如下代码就可以避免节点左右孩子翻转两次的情况:
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if(root==nullptr)
{
return root;
}
invertTree(root->left);
swap(root->left,root->right);
invertTree(root->left);
return root;
}
};
迭代
深度优先搜索(前序遍历)
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
stack<TreeNode*> az;
if(root==nullptr)
{
return root;
}
az.push(root);
while(!az.empty())
{
TreeNode* node=az.top();
az.pop();
swap(node->left,node->right);
if(node->right) az.push(node->right);
if(node->left) az.push(node->left);
}
return root;
}
};
如果对以上代码不理解,可以看下这篇博客力扣刷题之二叉树的前中后序遍历_skeet follower的博客-CSDN博客
广度优先搜索(前序遍历)
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;
}
};
如果对以上代码不理解,或者不清楚二叉树的层序遍历,可以看这篇层序遍历