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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/musechipin/article/details/85274891

题目描述: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;
        }  
    }
};


 

猜你喜欢

转载自blog.csdn.net/musechipin/article/details/85274891