Rabbet

悦已者

0%

翻转二叉树

发表于 更新于 分类于
本文字数： 853 阅读时长 ≈ 1 分钟

题目描述

  翻转一棵二叉树， 即左右节点互换

输入输出样例

1
2
3
4
5
6
Input:
     4
   /   \
  2     7
 / \   / \
1   3 6   9  
1
2
3
4
5
6
Output:
     4
   /   \
  7     2
 / \   / \
9   6 3   1

题解

  这是一道与树相关的算法，简单来说，我们应该先搞清楚当前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
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if (root == nullptr) return nullptr;

        invertTree(root->left);
        TreeNode* node = root->left;
        root->left = root->right;
        root->right = node;
        invertTree(root->right);

        return root;
    }
};