LeetCode114.二叉树展开为链表

I am jack

于 2021-03-18 11:15:25 发布

阅读量94

点赞数

分类专栏： # 树

本文链接：https://blog.youkuaiyun.com/qq_44932835/article/details/114968994

版权

树专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种将二叉搜索树扁平化的算法实现，通过递归和循环两种方式，将任意二叉树转化为单链表形式，便于进行后续的遍历操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述
一.递归

class Solution {
public:
    TreeNode* inOrder(TreeNode* root){
        if(root==NULL) return NULL;
        else{
            TreeNode* left = inOrder(root->left);
            TreeNode* right = inOrder(root->right);
            root->left = NULL;
            root->right = left;
            TreeNode *p = root;
            while(p->right!=NULL) p = p->right;
            p->right = right;
            return root;
        }
    }
    void flatten(TreeNode* root) {
        inOrder(root);
    }
};

二.循环

算法流程:
①若当前结点有左子树，将左孩子一直循环右结点插入到当前结点右结点
②否则一直循环右结点

class Solution {
public:
    void flatten(TreeNode* root) {
        while(root){
            if(root->left){
                TreeNode* p = root->left;
                while(p->right) p = p->right;
                p->right = root->right;
                root->right = root->left;root->left=NULL;
            }
            root = root->right;
        }
    }
};