LeetCode226 Invert Binary Tree

文章目录
  1. 1. 描述
  2. 2. 样例
  3. 3. 思路
  4. 4. 代码

描述

Invert a binary tree.

样例

1
2
3
4
5
6
7
8
9
10
11
12
13
14
Input:
4
/ \
2 7
/ \ / \
1 3 6 9

Output:

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

思路

翻转一棵二叉树。

dfs即可,每次swap左右子树。

代码

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 Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if (root == NULL) return NULL;
swap(root->left, root->right);
invertTree(root->left);
invertTree(root->right);
return root;
}
};
分享到 评论