将二叉树拆成链表

一.题目描述

将一棵二叉树按照前序遍历拆解成为一个假链表。所谓的假链表是说，用二叉树的 right指针，来表示链表中的 next 指针。

 注意事项

不要忘记将左儿子标记为 null，否则你可能会得到空间溢出或是时间溢出

样例

              1
               \
     1          2
    / \          \
   2   5    =>    3
  / \   \          \
 3   4   6          4
                     \
                      5
                       \
                        6

挑战 

不使用额外的空间耗费。

二.解题思路

先用递归的方法将左子树和右子树都变成假链表,再找到左子树的最后一个节点,令根节点的左子树为空,右子树等于左子树,再令左子树的最后一个节点的左子树为空,右子树等于根节点的右子树.

三.实现代码

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param root: a TreeNode, the root of the binary tree
     * @return: nothing
     */
    void flatten(TreeNode *root) {
        // write your code here
        if(root==NULL) return;
        TreeNode *left=root->left;
        TreeNode *right=root->right;
        TreeNode *p=new TreeNode;
        flatten(left);
        flatten(right);
        if(left!=NULL)
        { p=left;
          while(p->right!=NULL)
          { p=p->right;}
          root->left=NULL;
          root->right=left;
          p->left=NULL;
          p->right=right;
         
        }
         
    }
};

四.感悟

这道题的难点在于如何将左子树和右子树连起来,需要找到左子树的最后一个节点,将其与右子树的第一个节点连在一起.