leetcode114 二叉树展开为链表 中等

思路

原地转换成链表，即不创建额外的空间，直接对树的左右子树变换位置。
整体递归：

• 1、如果有left，先将left转成链表，再接入root的右边，并保存left的先序遍历的最后一个节点last；
• 2、如果有right，将right转成链表，并将转好的右子树链表接入last的右边；
  

/**
 * 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) {
        TreeNode* last=NULL;
        preoder(root,last);
    }
    void preoder(TreeNode* root,TreeNode *&last){
        if(root==NULL)
            return;
        if(!root->left&&!root->right){//如果节点左右子树都为空，则此节点为last
            last=root;
        }
        TreeNode* left=root->left;
        TreeNode* right=root->right;
        TreeNode* left_last=NULL;
        TreeNode* right_last=NULL;
        
        if(left){
            preoder(left,left_last);//若有左子树，递归将左子树转换为单链表
            root->left=NULL;
            root->right=left;
            last=left_last;
        }
        if(right){//right 是已经备份好的右子树
            preoder(right,right_last);//若有左子树，递归将左子树转换为单链表
            if(left_last){//如果有找到左子树的最后一个节点，则将右子树接入左子树的最后一个节点的后面
                left_last->right=right;
            }
            last=right_last;
        }
        
    }
};