力扣hot100——114.二叉树展开为链表

基于 Morris 遍历思想

将左子树插到右子树的位置,将原来的右子树插到左子树的最右结点,遍历右结点重复以上步骤,直至右结点为空。

class Solution {
public:
    void flatten(TreeNode* root) {
        if(root==nullptr) return;
        while(root){
            if(!root->left) root=root->right;
            else{
	            //找到左子树的最右结点
                TreeNode* node=root->left;
                while(node->right){
                    node=node->right;
                }
                node->right=root->right;
                root->right=root->left;
                root->left=nullptr;
                //遍历
                root=root->right;
            }
        }
    }
};
  • 时间复杂度:O(n)
  • 空间复杂度:O(1)

后序遍历

链表的顺序与先序遍历相同,但如果在先序遍历中修改指针指向会丢失原本的右孩子,无法完全遍历。那么不妨进行后序遍历,此时会先遍历右左孩子,然后更改指针指向。

class Solution {
public:
    TreeNode* pre=nullptr;
    void flatten(TreeNode* root) {
        if(root==nullptr) return;
        flatten(root->right);//遍历右子树
        flatten(root->left);//遍历左子树
        root->right=pre;//更改指向
        root->left=nullptr;
        pre=root;//更新pre
        }
};
  • 时间复杂度:O(n)
  • 空间复杂度:O(h)h为树的高度,递归调用栈
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值