题目描述:请完成一个函数,输入一个二叉树,该函数输出它的镜像。
示例 1: 输入:root = [4,2,7,1,3,6,9] 输出:[4,7,2,9,6,3,1]
方法一:递归法
根据二叉树镜像的定义,考虑递归遍历(dfs)二叉树,交换每个节点的左 / 右子节点,即可生成二叉树的镜像。
递归解析:
1、终止条件: 当节点 rootroot 为空时(即越过叶节点),则返回 nullnull ;
2、递推工作:
1、初始化节点 tmp ,用于暂存 root 的左子节点;
2、开启递归 右子节点 mirrorTree(root->right) ,并将返回值作为 root 的 左子节点 。
3、开启递归 左子节点 mirrorTree(tmp) ,并将返回值作为 root 的 右子节点 。
3、返回值: 返回当前节点 root;
class Solution {
public:
TreeNode* mirrorTree(TreeNode* root)
{
if(root == NULL) return root;
TreeNode* left = mirrorTree(root->left);
TreeNode* right = mirrorTree(root->right);
root->left = right;
root->right = left;
return root;
}
};
class Solution {
public:
TreeNode* mirrorTree(TreeNode* root)
{
if(root == NULL) return root;
//生成一个值为 root->val 的新节点
TreeNode* newroot = new TreeNode(root->val);
newroot->left = mirrorTree(root->right);
newroot->right = mirrorTree(root->left);
return newroot;
}
};
class Solution {
public:
TreeNode* mirrorTree(TreeNode* root) {
// 当节点为空时,直接返回
if(root == nullptr) return nullptr;
// 设置临时节点 tmp 来存储当前节点的左子树
TreeNode* tmp = root->left;
// 交换当前节点的左右子树
// 1.当前节点的左子树为节点的右子树,递归函数交换子树的左右子树
root->left = mirrorTree(root->right);
// 2.当前节点的右子树为节点的左子树,递归函数交换子树的左右子树
root->right = mirrorTree(tmp);
// 最后返回根节点
return root;
}
};
方法二:辅助栈(或队列)
利用栈(或队列)遍历树的所有节点 node ,并交换每个 node 的左 / 右子节点。
算法流程:
1、特例处理: 当 root 为空时,直接返回 null ;
2、初始化: 栈(或队列),本文用栈,并加入根节点 root 。
3、循环交换: 当栈 stack 为空时跳出;
1、出栈: 记为 node ;
2、添加子节点: 将 node 左和右子节点入栈;
3、交换: 交换 node 的左 / 右子节点。
4、返回值: 返回根节点 root 。
class Solution {
public:
TreeNode* mirrorTree(TreeNode* root) {
if(root == nullptr) return nullptr;
stack<TreeNode*> stack;
stack.push(root);
while (!stack.empty())
{
TreeNode* node = stack.top();
stack.pop();
if (node->left != nullptr) stack.push(node->left);
if (node->right != nullptr) stack.push(node->right);
TreeNode* tmp = node->left;
node->left = node->right;
node->right = tmp;
}
return root;
}
};