226. invert binary tree

Invert a binary tree.

1
2
3
4
5
     4
/
2 7
/ /
1 3 6 9

to

1
2
3
4
5
     4
/
7 2
/ /
9 6 3 1

1.递归

1
2
3
4
5
6
7
8
9
10
class Solution {
public:
TreeNode* (TreeNode* root) {
if (root == NULL)
return NULL;
TreeNode *tmp = root->left;
root->left = invertTree(root->right);
root->right = invertTree(tmp);
}
};

2.非递归

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
class Solution {
public:
TreeNode* (TreeNode* root) {
if (root == NULL)
return NULL;

queue<TreeNode *> p;
p.push(root);
while (! p.empty())
{
TreeNode *node = p.front();
TreeNode *left = node->left;
node->left = node->right;
node->right = left;
p.pop();

if (node->left != NULL)
p.push(node->left);
if (node->right != NULL)
p.push(node->right);
}
return root;
}
};