采用递归方法:依次从根节点开始交换左右子节点,递归到叶节点。
代码如下:
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
class Solution {
public:
void Mirror(TreeNode *pRoot) {
//递归写法
if(pRoot==NULL)
return;
TreeNode *tmp=pRoot->left;
pRoot->left=pRoot->right;
pRoot->right=tmp;
Mirror(pRoot->left);
Mirror(pRoot->right);
}
};
非递归写法思路如下:从根节点开始入栈,若栈非空,则取出栈顶元素,交换该元素节点的左右子节点,并且判断该元素的左右子节点是否非空,若非空,则将其压入栈中,如此循环,直到栈内元素全部弹出。
代码如下(未经验证,只看逻辑)
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
void Mirror(TreeNode *pRoot) {
//非递归
if(pRoot==NULL)
return;
stack<TreeNode*> s1;
s1.push(pRoot);
while(!s1.empty())
{
TreeNode *root=s1.top();
s1.pop();
if(root->left!=NULL||root->right!=NULL)
{
TreeNode *tmp=root->left;
root->left=root->right;
root->right=tmp;
}
if(root->right!=NULL)
{
s1.push(root->right);
}
if(root->left!=NULL);
{
s1.push(root->left);
}
}
}
};