二叉树展成链表（C++）

给定一个二叉树，原地将它展开为链表。

例如，给定二叉树

这道题的题眼是愿地将二叉树展成链表，注意是原地，大家仔细看就会发现这个用前序遍历加vector的方法很容易将这一题求解，代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void flatten(TreeNode* root) {
        //二叉树展成链表，我们看一下链表结点顺序，
        //发现他们是二叉树的前序遍历的顺序
        //那么如果我们不是原地将它们展成链表；
        //就可以设计一个vector来装前序遍历的节点
        //最后将它们给连起来
        //判空
        if(!root){
            return;
        }
        vector<TreeNode*>res;
        preorder(root,res);
        for(int i=0;i<res.size()-1;i++){
            res[i]->left=NULL;
            res[i]->right = res[i+1];
        }
        return;
    }
    void preorder(TreeNode* root, vector<TreeNode*> &res){
        if(!root){
            return;
        }
        res.push_back(root);
        preorder(root->left,res);
        preorder(root->right,res);
    }
};

但是，我们都知道这是不符合要求的，所以，下面的思路才是正确的
下面介绍标准的in-place解法：

class Solution {
public:
    void flatten(TreeNode* root) {
        if(!root){
            return;
        }
        //最简单的解法来了；
        //就是按照题目要求，将右子树的节点一个一个安在左子树上然后
        //将左子树接在root的右子树上
        while(root){
            //如果左子树不为空，先找左子树的最右节点
            if(root->left){
            auto left_right = root->left;
            //最右节点
            while(left_right->right){
                left_right = left_right->right;
            }
            //将root的右孩子放到最右节点子树上；
            left_right->right = root->right;
            //root的右孩子换成root的左孩子
            root->right = root->left;
            //root的左孩子置空
            root->left = NULL;
            }
            //继续下一个节点
            root=root->right;
        }
        return;
    }

};

基本思路：前序遍历每个节点，先调用左结点，左结点返回后，如果该节点有右子树，就把右子树接到左子树的末尾，这时该根节点的整个子树就成为了单链；因为要右子树移接到左子树的末尾，所以要先有个指针last始终指向左子树的叶节点（即最末节点），移接结束后，根节点的u右子树要z置空，last更新到新左子树末尾。递归调用结束后，返回的是单链但未必每个节点都是左孩子为空（题目要求），所以还需要左转右，左置空。

class Solution {
public:
    void flatten(TreeNode* root) {
        if(!root){
            return;
        }
        TreeNode* last=root;
        //找最右边的叶节点
        while(last->left||last->right){
            if(last->left){
                last=last->left;
            }
            else if(last->right){
                last=last->right;
            }  
        }
        //cout<<last->val<<endl;
        aftorder(root, last);
        last=root;
        while(last){
            if(last->left){
                last->right=last->left;
                last->left=NULL;
            }
            last=last->right;
        }
       return; 
    }
private:
    void aftorder(TreeNode* root,TreeNode*&last){
        if(!root){
            return;
        }
        //访问左子树
        aftorder(root->left, last);
        //访问右子树前先进行移花接木，将右子树接到last上
        if(root->right&&root->left){
            last->right=root->right;
            root->right=NULL;
            aftorder(last->right, last);//注意这里并不是访问右子树
        }
        else{
            //访问右子树
            aftorder(root->right, last);  
        } 
        //更新last
        while(last->left||last->right){
            if(last->left){
                last=last->left;
            }
            else if(last->right){
                last=last->right;
            }  
        }
        cout<<last->val<<endl;
    }
};