题目描述:
翻转一棵二叉树。
示例:
输入:
4
/ /
2 7
/ / / /
1 3 6 9
输出:
4
/ /
7 2
/ / / /
9 6 3 1
思路分析:
要想交换一个二叉树,只需要在遍历每个节点的时候,交换其左右两个孩子的位置,即可完成对整颗二叉树的逆置。
注意,此题不可以使用中序遍历来的反转二叉树。
递归法:
通过先序遍历每个节点,然后交换左右两个孩子,来实现对整个二叉树的翻转。
代码:
TreeNode* invertTree(TreeNode* root) {
if(root==NULL) return root; //递归终止条件
swap(root->left,root->right); //中,
invertTree(root->left); //左
invertTree(root->right); //右
return root;
DFS翻转二叉树
代码:
TreeNode* invertTree(TreeNode* root) {
if(root == NULL) return root;
stack<TreeNode*> sta; //深度遍历需要借助栈
sta.push(root); //先把头节点放到栈中去;
while(!sta.empty())
{
TreeNode* node=sta.top(); //获取到栈顶节点,然后弹出
sta.pop();
swap(node->left,node->right); //交换栈顶节点左右孩子
if(node->right) sta.push(node->right); //先压右孩子,因为栈是先进后出
if(node->left) sta.push(node->left); //在压左孩子
}
return root;
}
层序优先遍历反转二叉树
代码:
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;
}
总结:
其实对于反转二叉树,最核心的思想便是遍历二叉树的每个节点,然后交换左右两个孩子的位置即可。