翻转二叉树
问题:
思路:
1.层次遍历(迭代)
利用队列遍历,利用STL中的Swap函数将出队的树节点的指向左右子树根节点的指针交换,并将指向左右子树根节点的指针入队
遍历开始前将根节点入队,遍历结束条件为队列为空
2.分治递归
将一颗树翻转,即将其左右子树交换,再将左右子树翻转
当树为空时,递归函数直接返回
代码:
层次遍历(迭代):
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if( root==NULL )
return root;
queue<TreeNode*> q;
TreeNode* p;
q.push( root );
while( q.size()!=0 )
{
p=q.front();
q.pop();
swap( p->left,p->right );
if( p->left!=NULL )
q.push( p->left );
if( p->right!=NULL )
q.push( p->right );
}
return root;
}
};
分治递归:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
change( root );
return root;
}
void change( TreeNode* p )
{
if( p==NULL )
return;
swap( p->left,p->right );
change( p->left );
change( p->right );
}
};