LeetCode: Flatten Binary Tree to Linked List

最新推荐文章于 2024-01-17 15:25:50 发布

原创最新推荐文章于 2024-01-17 15:25:50 发布 · 535 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

144 篇文章

订阅专栏

本文介绍了一种将二叉树扁平化的算法实现，通过递归方式将树的左子树作为右子树的最右侧节点的子节点，最终使整个二叉树转化为只有右子树的链表形式。

思路：将树的子树插入原有的右子树之前，组合成新的右子树，左子树和右子树依次这样递归求解。

code:

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