114. Flatten Binary Tree to Linked List

最新推荐文章于 2022-07-27 15:08:45 发布

原创最新推荐文章于 2022-07-27 15:08:45 发布 · 206 阅读

0 ·

CC 4.0 BY-SA版权

leetcode-二叉树专栏收录该内容

9 篇文章

订阅专栏

本文介绍二叉树扁平化的两种方法：一种利用vector进行先序遍历，空间复杂度为O(n)；另一种采用直接修改指针的方式实现，无额外空间开销。通过这两种方式可以将任意二叉树转换成单链表形式。

方法1:

使用vector 保存遍历节点，空间复杂度O(n)

class Solution {
public:
    void PreOrder(TreeNode* root,vector<TreeNode*> &vt)
    {
        if(root==NULL)
            return ;
        vt.push_back(root);
        PreOrder(root->left,vt);
        PreOrder(root->right,vt);
    }
    
    void flatten(TreeNode* root) {
        if(!root)
            return ;
        vector<TreeNode*> vt;
        PreOrder(root, vt);
        for(int i=0;i<vt.size()-1;i++)
        {
            vt[i]->left=NULL;
            vt[i]->right=vt[i+1];
        }
    }
};

方法2:

根据先序遍历，将当前节点的右子树连接到左子树的最右节点。无空间复杂度

class Solution {
public:

    void flatten(TreeNode* root) {
        if(!root)
            return ;
        TreeNode* now=root;
        while(now)
        {
            if(now->left)
            {
                TreeNode* pre=now->left;
                while(pre->right)
                    pre=pre->right;
                pre->right=now->right;
                now->right=now->left;
                now->left=NULL;
            }
            now=now->right;
        }
    }
};