114. Flatten Binary Tree to Linked List的C++解法

最新推荐文章于 2020-06-07 20:07:19 发布

原创最新推荐文章于 2020-06-07 20:07:19 发布 · 244 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

158 篇文章

订阅专栏

本文介绍了一种将二叉树结构转换为链表的方法，重点在于找到左右子树的最右下角节点，以便正确连接子树。通过递归或迭代的方式，将左子树作为右子树的一部分，最终实现整个二叉树的链表化。

题目描述：https://leetcode.com/problems/flatten-binary-tree-to-linked-list/

这个题的关键是找到左右子树的最右下角的位置，因为这将是下一次接子树的起点。下面这个方法用了递归，假设左右子树将在helper中接好，并且helper返回的是接好的子树的最右下角的位置。

class Solution {
public:
    void flatten(TreeNode* root) {
        helper(root);
    }
    TreeNode* helper(TreeNode* root){
        if (root==NULL) return NULL;
        TreeNode* left=helper(root->left);// left记录了root->left的最右下角的位置，用于接上root->right
        TreeNode* right=helper(root->right);//right 记录了root->right最右下角的位置，拼接完成后这将是root的最右下角
        if (left!=NULL) 
        {
            left->right=root->right;//将右子树接到左子树最下面
            root->right=root->left;//将整棵左子树接到右子树的位置
            root->left=NULL;
         }
        if (right!=NULL) return right;//右子树存在最右下角的位置则优先返回这个位置（因为这个位置将是最右的）
        if (left!=NULL) return left;//如果没有右子树的右下角，则左子树的右下角是最后一个位置
        return root;//都没有就返回根
    }
};

更好的做法：
只有左边还有子树，就用一个指针遍历到左边子树的最最右边到底的一个节点，然后把root右边子树全部挂在这个节点上，然后再通过root->right = root -> left; root -> left = NULL; 消灭右边子树，然后root = root-> right;循环往复，不断消灭右边子树。

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