题目描述:
操作给定的二叉树,将其变换为源二叉树的镜像。
输入描述:
解题思路:总共有三种方法,非递归有队列、栈两种方法。
非递归:栈实现,第一步:根节点不为空将根节点压入栈中。第二步:将栈顶的元素出栈,赋值给p,若p的左右结点不都为空,则交换左右子树的值。第三步:若该结点的左子树不为空则将该节点的左子树压入栈中,若结点的右子树不为空将右子树压入栈中。 第四步:一直循环,直到栈为空
class Solution {
public:
void Mirror(TreeNode *pRoot)
{
if(NULL == pRoot)
return;
stack<TreeNode*> stk;
stk.push(pRoot);
while(! stk.empty())
{
TreeNode* p = stk.top();
stk.pop();
if(p->left != NULL || p->right != NULL)
{
TreeNode* tmp = p->left;
p->left = p->right;
p->right = tmp;
}
if(p->left)
stk.push(p->left);
if(p->right)
stk.push(p->right);
}
}
};
非递归:队列实现,根栈实现的方法差不多。
class Solution {
public:
void Mirror(TreeNode *pRoot)
{
if(NULL == pRoot)
return;
queue<TreeNode*> stk;
stk.push(pRoot);
while(! stk.empty())
{
TreeNode* p = stk.front();
stk.pop();
if(p->left != NULL || p->right != NULL)
{
TreeNode* tmp = p->left;
p->left = p->right;
p->right = tmp;
}
if(p->left)
stk.push(p->left);
if(p->right)
stk.push(p->right);
}
}
};
递归:先前序遍历这棵树的每个结点,如果遍历到的结点有子结点,就交换它的两个子节点,当交换完所有的非叶子结点的左右子结点之后,就得到了树的镜像
class Solution {
public:
void Mirror(TreeNode *pRoot)
{
if(NULL == pRoot)
return;
TreeNode* tmp = pRoot->left;
pRoot->left = pRoot->right;
pRoot->right = tmp;
Mirror(pRoot->left);
Mirror(pRoot->right);
}
};