编程训练第八十期——翻转二叉树

编程问题:

翻转一棵二叉树。


示例:

  • 输入:[4,2,7,1,3,6,9]
    输出:[4,7,2,9,6,3,1]

解法:

1.递归
显然,我们从根节点开始,递归地对树进行遍历,并从叶子结点先开始翻转。如果当前遍历到的节点 root 的左右两棵子树都已经翻转,那么我们只需要交换两棵子树的位置,即可完成以root 为根节点的整棵子树的翻转。
时间复杂度O(N)
空间复杂度O(N),使用的空间由递归栈的深度决定,它等于当前节点在二叉树中的高度。在平均情况下,二叉树的高度与节点个数为对数关系,即 O(logN)。而在最坏情况下,树形成链状,空间复杂度为 O(N)。

/**
 * 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) return root;
        TreeNode *leftNode = invertTree(root->left);
        TreeNode *rightNode = invertTree(root->right);
        root->left = rightNode;
        root->right = leftNode;
        return root;
    }
};

同样采用递归的方法,采用"判断对称二叉树”的思想,建立一颗与root镜像对称的二叉树,在实现时,以root的左结点建立newRoot右结点,以以root的右结点建立newRoot左结点。
时间复杂度O(N)
空间复杂度O(N)

/**
 * 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 {
private:
    void reverse(TreeNode *root, TreeNode *newRoot)
    {
        if (!root)
            return;
        if (root->left)
        {
            TreeNode *rightNode = new TreeNode(root->left->val);
            newRoot->right = rightNode;
            reverse(root->left, rightNode);
        }
        if (root->right)
        {
            TreeNode *leftNode = new TreeNode(root->right->val);
            newRoot->left = leftNode;
            reverse(root->right, leftNode);
        }

    }

public:
    TreeNode* invertTree(TreeNode* root) {
        if (!root) return root;
        TreeNode *newRoot = new TreeNode(root->val);
        reverse(root, newRoot);
        return newRoot
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值