[leetcode]Flatten Binary Tree to Linked List

最新推荐文章于 2025-08-13 10:25:05 发布

weixin_34072637

最新推荐文章于 2025-08-13 10:25:05 发布

阅读量50

点赞数

文章标签：数据结构与算法 c/c++

本文介绍了一种算法，该算法通过后序遍历的方式将二叉树转换为前序遍历的链表。文章详细展示了如何递归地处理二叉树的左右子树，并调整指针以确保转换后的链表符合要求。

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

class Solution {
public:
    //目的是将binary tree变成前序遍历的list，但是变换过程是后序遍历
    void flatten(TreeNode *root, TreeNode *&first, TreeNode *&last){
        if (!root)
            return ;

        TreeNode *leftFirst = NULL;
        TreeNode *leftLast = NULL;
        TreeNode *rightFirst = NULL;
        TreeNode *rightLast = NULL;
        flatten(root->left, leftFirst, leftLast);        //对于向下调用问题，递归的先处理子树
        flatten(root->right, rightFirst, rightLast);    //对于向下调用问题，递归的先处理子树

        first = root;                                    //对于向上返回问题，first肯定是root
        if (root->right)                                //如果有右子树
            last = rightLast;
        else if (root->left)                            //没有右子树但是有左子树
            last = leftLast;
        else                                            //没有子树
            last = root;

        root->left = NULL;
        if (leftFirst && !rightFirst){
            root->right = leftFirst;
        }
        else if (!leftFirst && rightFirst){
            root->right = rightFirst;
        }
        else if (leftFirst && rightFirst){
            root->right = leftFirst;
            leftLast->right = rightFirst;
        }
        else{
            root->right = NULL;
        }
    }

    void flatten(TreeNode *root) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        if (!root)
            return ;

        TreeNode *first;
        TreeNode *last;
        flatten(root, first, last);

    }
};

EOF