题目描述
Invert a binary tree.
=>
题目要求将二叉树的左右子树进行逆转。
解题思路
根据题目的要求可以进行一层一层的逆转,也就是交换结点的左右子树。将未处理的元素放入栈中,本题的关键在于对于元素的入栈顺序进行控制。
/**
* 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 NULL;
TreeNode* head = root;
stack<TreeNode*> nodes;
nodes.push(head);
while(!nodes.empty())
{
TreeNode* r = nodes.top();
nodes.pop();
swap(r->left, r->right);
if(r->left != NULL) nodes.push(r->left);
if(r->right != NULL) nodes.push(r->right);
}
return root;
}
};