leetcode226:反转二叉树

摘要:反转二叉树。

解法一:递归

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class {
public:
TreeNode* invertTree(TreeNode* root) {
if(!root) return NULL;
TreeNode* t = root -> left;
root -> left = invertTree(root -> right);
root -> right = invertTree(t);
return root;
}
};

解法二:非递归

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class {
public:
TreeNode* invertTree(TreeNode* root) {
if(!root) return NULL;
queue <TreeNode*> q;
q.push(root);
while(!q.empty()){
TreeNode* node = q.front();
q.pop();
TreeNode* tmp = node -> left;
node -> left = node -> right;
node -> right = tmp;
if(node -> left) q.push(node -> left);
if(node -> right) q.push(node -> right);
}
return root;
}
};